diff --git a/bot/code_generator.py b/bot/code_generator.py
index acf37fd..c91104f 100644
--- a/bot/code_generator.py
+++ b/bot/code_generator.py
@@ -3,9 +3,7 @@
 import traceback
 import datetime
 import py_compile
-import threading
 import logging
-import json
 import black
 from flask import Flask, request, jsonify
 from groq import Groq
@@ -78,10 +76,10 @@ def extract_code(self, response):
         """Extract the code within the triple quotes."""
         parts = response.split("```")
         if len(parts) >= 2:
-            code = parts[1].strip()
+            code = parts[1]
+            code = "\n".join(code.splitlines()[1:])
             return code_formatter(code)
         return ""
-
 class DebugAgent:
     def __init__(self, workspace):
         self.workspace = workspace