diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index e7186fc..4de8412 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -174,10 +174,12 @@ async def stream_code(websocket: WebSocket):
         try:
             if params.get("resultImage") and params["resultImage"]:
                 prompt_messages = assemble_prompt(
-                    params["image"], valid_stack, params["resultImage"]
+                    params["images"], generated_code_config, params["resultImage"]
                 )
             else:
-                prompt_messages = assemble_prompt(params["image"], valid_stack)
+                prompt_messages = assemble_prompt(
+                    params["images"], generated_code_config
+                )
         except:
             await websocket.send_json(
                 {