diff --git a/reflex/components/datadisplay/shiki_code_block.py b/reflex/components/datadisplay/shiki_code_block.py
index 9b2be4beb..31931b9ed 100644
--- a/reflex/components/datadisplay/shiki_code_block.py
+++ b/reflex/components/datadisplay/shiki_code_block.py
@@ -784,8 +784,8 @@ class ShikiHighLevelCodeBlock(ShikiCodeBlock):
                 code, StringVar(_js_expr=r"/\/\/ \[!code.*?\]/g", _var_type=str), ""
             )
         if isinstance(code, str):
-            cleaned_code = re.sub(r"// \[!code.*?\]", "", code)
-            return cleaned_code
+            return re.sub(r"// \[!code.*?\]", "", code)
+
 
 
 class TransformerNamespace(ComponentNamespace):