diff --git a/tinygrad/viz/index.html b/tinygrad/viz/index.html
index 6eb0630347..f1be9687f1 100644
--- a/tinygrad/viz/index.html
+++ b/tinygrad/viz/index.html
@@ -434,9 +434,15 @@
     }
     if (event.key == "Enter") {
       event.preventDefault()
+      if (currentKernel === -1) {
+        currentKernel = 0;
+        expandKernel = true;
+      }
+      else {
+        expandKernel = !expandKernel;
+      }
       currentUOp = 0;
       currentRewrite = 0;
-      expandKernel = !expandKernel;
       main();
     }
     if (event.key == "ArrowUp") {