diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index 9a1b2c0..1d75ca1 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -76,7 +76,7 @@ async def stream_code(websocket: WebSocket):
     valid_stack = cast(Stack, generated_code_config)
 
     # Validate the input mode
-    input_mode = params.get("inputMode")
+    input_mode = params.get("inputMode", "image")
     if not input_mode in get_args(InputMode):
         await throw_error(f"Invalid input mode: {input_mode}")
         raise Exception(f"Invalid input mode: {input_mode}")