diff --git a/kernel/model/block.go b/kernel/model/block.go
index 6f4ccf155..9ebe50a33 100644
--- a/kernel/model/block.go
+++ b/kernel/model/block.go
@@ -741,7 +741,11 @@ func GetHeadingChildrenDOM(id string, removeFoldAttr bool) (ret string) {
 			return ast.WalkContinue
 		})
 
-		child.SetIALAttr("parent-heading", id)
+		if removeFoldAttr {
+			child.RemoveIALAttr("parent-heading")
+		} else {
+			child.SetIALAttr("parent-heading", id)
+		}
 	}
 
 	if removeFoldAttr {