diff --git a/layouts/partials/highlight-js.html b/layouts/partials/highlight-js.html
index 09d7bc3..7026b12 100644
--- a/layouts/partials/highlight-js.html
+++ b/layouts/partials/highlight-js.html
@@ -1,19 +1,19 @@
 {{ if .Site.Params.highlightjs }}
     {{ if or (not (isset .Params "highlight")) (.Params.highlight) }}
-        {{ .Scratch.Set "hl_languages" (union (.Site.Params.highlightjslanguages) (.Params.highlightjslanguages)) }}
+        {{ $.Scratch.Set "hl_languages" (union (.Site.Params.highlightjslanguages) (.Params.highlightjslanguages)) }}
         <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/9.12.0/highlight.min.js"></script>
-        {{ with .Scratch.Get "hl_languages" }}
+        {{ with $.Scratch.Get "hl_languages" }}
             {{ range . }}
                 <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/9.12.0/languages/{{.}}.min.js"></script>
             {{ end }}
             
         {{ end }}
         <script type="text/javascript">
-            {{ with .Scratch.Get "hl_languages" }}
+            {{ with $.Scratch.Get "hl_languages" }}
             hljs.configure({languages: [{{(delimit . ", ")}}]});
             {{ end }}
             hljs.initHighlightingOnLoad();
         </script>
-        {{ .Scratch.Delete "hl_languages" }}
+        {{ $.Scratch.Delete "hl_languages" }}
     {{ end }}    
 {{ end }}