diff --git a/src/components/__tests__/__snapshots__/Highlight.test.js.snap b/src/components/__tests__/__snapshots__/Highlight.test.js.snap
index ff205de..fa0b8ca 100644
--- a/src/components/__tests__/__snapshots__/Highlight.test.js.snap
+++ b/src/components/__tests__/__snapshots__/Highlight.test.js.snap
@@ -114,7 +114,12 @@ exports[`<Highlight /> snapshots renders correctly 1`] = `
       <span
         class="token plain"
       >
-          console
+          
+      </span>
+      <span
+        class="token console class-name"
+      >
+        console
       </span>
       <span
         class="token punctuation"
@@ -123,7 +128,7 @@ exports[`<Highlight /> snapshots renders correctly 1`] = `
         .
       </span>
       <span
-        class="token function"
+        class="token method function property-access"
         style="color: rgb(154, 134, 253);"
       >
         log
@@ -216,7 +221,7 @@ exports[`<Highlight /> snapshots renders correctly 1`] = `
         class="token plain"
       />
       <span
-        class="token keyword"
+        class="token keyword control-flow"
         style="color: rgb(255, 204, 153);"
       >
         return
@@ -244,7 +249,7 @@ exports[`<Highlight /> snapshots renders correctly 1`] = `
          
       </span>
       <span
-        class="token operator"
+        class="token arrow operator"
         style="color: rgb(224, 145, 66);"
       >
         =&gt;