diff --git a/main.js b/main.js
index f0468c8..49837d5 100644
--- a/main.js
+++ b/main.js
@@ -262,11 +262,11 @@ function update() {
   }
 
   else if (state === 'player-win') {
-    //
+    idle = true;
   }
 
   else if (state === 'game-over') {
-    //
+    idle = true;
   }
 
   else {