diff --git a/api/main.py b/api/main.py
index 91661131faac946534bd67e478c80726ace517c0..007d38a92eab849c79f53cdbac09a0ec67e700a5 100644
--- a/api/main.py
+++ b/api/main.py
@@ -77,7 +77,11 @@ if DEBUG:
             headers=request.headers,
             stream=True,
         )
-        return Response(content=response.content, status_code=response.status_code)
+        return Response(
+            content=response.raw.read(),
+            status_code=response.status_code,
+            headers=dict(response.headers.items()),
+        )
 
 else:
     app.mount("/", StaticFiles(directory="frontend/build/", html=True), name="frontend")
diff --git a/api/optimization.py b/api/optimization.py
index 39cf021cd38760bc6a7dfc874e8350d8773d706e..11198108a5822f62ddf84ab4ce9d5af223619afe 100644
--- a/api/optimization.py
+++ b/api/optimization.py
@@ -9,8 +9,8 @@ from task import SentimentAnalysis
 #     global optimizer
 #     optimizer = Optimizer(debug=DEBUG)
 
-_evolution_model: LLMModel = None
-_evaluation_model: LLMModel = None
+_evolution_model: LLMModel | None = None
+_evaluation_model: LLMModel | None = None
 # _model_call_type: get_type_hints(LLMModel).get("__call__")
 
 
@@ -82,10 +82,7 @@ class MultiProcessOptimizer:
 
         # currently fix task
         options = Namespace(use_grammar=False, debug=self.debug)
-        task = SentimentAnalysis(
-            evaluation_model,
-            options
-        )
+        task = SentimentAnalysis(evaluation_model, options)
 
         optimizer_class = GeneticAlgorithm
         # optimizer_class = DifferentialEvolution