diff --git a/reflex/.templates/apps/demo/code/pages/datatable.py b/reflex/.templates/apps/demo/code/pages/datatable.py
index 4cc9d39ea..2a7e79c93 100644
--- a/reflex/.templates/apps/demo/code/pages/datatable.py
+++ b/reflex/.templates/apps/demo/code/pages/datatable.py
@@ -306,7 +306,7 @@ def datatable_page() -> rx.Component:
                 ),
                 rx.chakra.tab_panels(
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             code_show,
                             language="python",
                             show_line_numbers=True,
@@ -316,7 +316,7 @@ def datatable_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             data_show,
                             language="python",
                             show_line_numbers=True,
@@ -326,7 +326,7 @@ def datatable_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             state_show,
                             language="python",
                             show_line_numbers=True,
@@ -336,7 +336,7 @@ def datatable_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             darkTheme_show,
                             language="python",
                             show_line_numbers=True,
diff --git a/reflex/.templates/apps/demo/code/pages/forms.py b/reflex/.templates/apps/demo/code/pages/forms.py
index 5b39ba5c1..38efc3e0f 100644
--- a/reflex/.templates/apps/demo/code/pages/forms.py
+++ b/reflex/.templates/apps/demo/code/pages/forms.py
@@ -150,7 +150,7 @@ def forms_page() -> rx.Component:
                 ),
                 rx.chakra.tab_panels(
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             forms_1_code,
                             language="python",
                             show_line_numbers=True,
@@ -160,7 +160,7 @@ def forms_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             forms_1_state,
                             language="python",
                             show_line_numbers=True,
@@ -223,7 +223,7 @@ def forms_page() -> rx.Component:
                 ),
                 rx.chakra.tab_panels(
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             forms_2_code,
                             language="python",
                             show_line_numbers=True,
@@ -233,7 +233,7 @@ def forms_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             forms_2_state,
                             language="python",
                             show_line_numbers=True,
diff --git a/reflex/.templates/apps/demo/code/pages/graphing.py b/reflex/.templates/apps/demo/code/pages/graphing.py
index 6b5defb46..3eeef92c1 100644
--- a/reflex/.templates/apps/demo/code/pages/graphing.py
+++ b/reflex/.templates/apps/demo/code/pages/graphing.py
@@ -144,7 +144,7 @@ def graphing_page() -> rx.Component:
                 ),
                 rx.chakra.tab_panels(
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             graph_1_code,
                             language="python",
                             show_line_numbers=True,
@@ -154,7 +154,7 @@ def graphing_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             data_1_show,
                             language="python",
                             show_line_numbers=True,
@@ -217,7 +217,7 @@ def graphing_page() -> rx.Component:
                 ),
                 rx.chakra.tab_panels(
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             graph_2_code,
                             language="python",
                             show_line_numbers=True,
@@ -227,7 +227,7 @@ def graphing_page() -> rx.Component:
                         padding_y=".25em",
                     ),
                     rx.chakra.tab_panel(
-                        rx.chakra.code_block(
+                        rx.code_block(
                             graph_2_state,
                             language="python",
                             show_line_numbers=True,