diff --git a/layouts/_default/_markup/render-codeblock-math.html b/layouts/_default/_markup/render-codeblock-math.html
index 29d2a597e1..566b662035 100644
--- a/layouts/_default/_markup/render-codeblock-math.html
+++ b/layouts/_default/_markup/render-codeblock-math.html
@@ -1,5 +1,5 @@
 {{- partial "shortcodes/math.html" (dict
-  "context" .
+  "context" .Page
   "content" .Inner
   "align"  (index .Attributes "align")
 ) }}
\ No newline at end of file
diff --git a/layouts/_default/_markup/render-codeblock-mermaid.html b/layouts/_default/_markup/render-codeblock-mermaid.html
index b3ba7d7973..196cac09f1 100644
--- a/layouts/_default/_markup/render-codeblock-mermaid.html
+++ b/layouts/_default/_markup/render-codeblock-mermaid.html
@@ -1,5 +1,5 @@
 {{- partial "shortcodes/mermaid.html" (dict
-  "context" .
+  "context" .Page
   "content" .Inner
   "align"  (index .Attributes "align")
 ) }}
\ No newline at end of file