diff --git a/layouts/partials/search/functions/parse-headings.html b/layouts/partials/search/functions/parse-headings.html
index d54f1894876..db48e09abee 100644
--- a/layouts/partials/search/functions/parse-headings.html
+++ b/layouts/partials/search/functions/parse-headings.html
@@ -1,8 +1,10 @@
 {{- $headings := slice }}
-{{- range .Fragments.HeadingsMap }}
-  {{- $headings = $headings | append (dict
-    "anchor" .ID
-    "title" (.Title | plainify))
-  }}
+{{- with .Fragments }}
+  {{- range .HeadingsMap }}
+    {{- $headings = $headings | append (dict
+      "anchor" .ID
+      "title" (.Title | plainify))
+    }}
+  {{- end }}
 {{- end }}
 {{- return $headings -}}