diff --git a/reflex/.templates/web/components/shiki/code.js b/reflex/.templates/web/components/shiki/code.js
index f4535cdd5..322096563 100644
--- a/reflex/.templates/web/components/shiki/code.js
+++ b/reflex/.templates/web/components/shiki/code.js
@@ -1,23 +1,23 @@
 import { useEffect, useState } from "react"
 import { codeToHtml} from "shiki"
 
-export function Code ({code, theme, language, themes, transformers}) {
+export function Code ({code, theme, language, transformers, ...divProps}) {
     const [codeResult, setCodeResult] = useState("")
     useEffect(() => {
         async function fetchCode() {
           const result = await codeToHtml(code, {
-            lang: language || "plaintext",
-            theme: theme || "nord",
-            transformers: transformers || []
+            lang: language,
+            theme,
+            transformers
           });
           setCodeResult(result);
         }
         fetchCode();
-      }, [code, language, theme, themes, transformers]
+      }, [code, language, theme, transformers]
 
     )
-
+    console.log(divProps)
     return (
-        <div dangerouslySetInnerHTML={{__html: codeResult}}></div>
+        <div dangerouslySetInnerHTML={{__html: codeResult}} {...divProps}  ></div>
     )
 }