diff --git a/server/app.py b/server/app.py
index d2bb93e..18c4899 100644
--- a/server/app.py
+++ b/server/app.py
@@ -49,7 +49,9 @@ def setup():
 
 @app.route('/get_updated_sentence')
 def get_updated_sentence():
-    return jsonify(generator.get_status())
+    if not generator:
+        return jsonify({"error": "Generator not initialized"}), 400
+    return generator.get_status()
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', threaded=True, debug=True, port=args.port)
\ No newline at end of file
diff --git a/server/sources/generator.py b/server/sources/generator.py
index aff7872..0c70f06 100644
--- a/server/sources/generator.py
+++ b/server/sources/generator.py
@@ -41,7 +41,6 @@ class GeneratorLLM():
     
     def get_status(self) -> dict:
         with self.state.lock:
-            print("Returning status:\n", self.state.status())
             return jsonify(self.state.status())
 
     @abstractmethod
diff --git a/sources/llm_provider.py b/sources/llm_provider.py
index 27bbdb6..c8f7f88 100644
--- a/sources/llm_provider.py
+++ b/sources/llm_provider.py
@@ -121,7 +121,6 @@ class Provider:
             is_complete = False
             while not is_complete:
                 response = requests.get(f"http://{self.server_ip}/get_updated_sentence")
-                print("raw:", response.json())
                 thought = response.json()["sentence"]
                 is_complete = bool(response.json()["is_complete"])
                 time.sleep(2)