diff --git a/lib/widgets/chat_page/code_block.dart b/lib/widgets/chat_page/code_block.dart
index 2b7ee57..f407f75 100644
--- a/lib/widgets/chat_page/code_block.dart
+++ b/lib/widgets/chat_page/code_block.dart
@@ -27,12 +27,10 @@ class CodeBlock extends StatelessWidget {
                       style: TextStyle(fontFamily: "monospace"),
                     ),
                   ),
-                  TextButton(
+                  TextButton.icon(
                     onPressed: () {},
-                    child: Row(
-                      spacing: 4,
-                      children: [Icon(Icons.copy), Text("Copy")],
-                    ),
+                    icon: Icon(Icons.copy),
+                    label: Text("Copy"),
                   ),
                 ],
               ),