diff --git a/src/lib/Code.svelte b/src/lib/Code.svelte
index 57e08ec..8cb3dda 100644
--- a/src/lib/Code.svelte
+++ b/src/lib/Code.svelte
@@ -55,6 +55,8 @@
 {:else}
   <div class="code-block is-relative">
     <button class="button is-light is-outlined is-small p-2" on:click={copyFunction}>Copy</button>
-    <HighlightAuto code={text} {language} />
+    <HighlightAuto code={text} let:highlighted>
+      <LineNumbers {highlighted} />
+    </HighlightAuto>
   </div>
 {/if}