diff --git a/frontend/src/generateCode.ts b/frontend/src/generateCode.ts
index 2ad578f..7f0c0b6 100644
--- a/frontend/src/generateCode.ts
+++ b/frontend/src/generateCode.ts
@@ -55,6 +55,7 @@ export function generateCode(
     } else {
       console.error("WebSocket error code", event);
       toast.error(ERROR_MESSAGE);
+      onComplete();
     }
   });