diff --git a/packages/extension-task-item/src/task-item.ts b/packages/extension-task-item/src/task-item.ts
index 5299b70f..3183e043 100644
--- a/packages/extension-task-item/src/task-item.ts
+++ b/packages/extension-task-item/src/task-item.ts
@@ -111,6 +111,7 @@ export const TaskItem = Node.create<TaskItemOptions>({
         }
       })
 
+      listItem.dataset.checked = node.attrs.checked
       if (node.attrs.checked) {
         checkbox.setAttribute('checked', 'checked')
       }
@@ -132,6 +133,7 @@ export const TaskItem = Node.create<TaskItemOptions>({
             return false
           }
 
+          listItem.dataset.checked = updatedNode.attrs.checked
           if (updatedNode.attrs.checked) {
             checkbox.setAttribute('checked', 'checked')
           } else {