diff --git a/tinygrad/viz/index.html b/tinygrad/viz/index.html
index e8a34d89d9..0310c6428b 100644
--- a/tinygrad/viz/index.html
+++ b/tinygrad/viz/index.html
@@ -102,10 +102,10 @@
     fill: none;
     stroke-width: 1.4px;
   }
-  .highlight rect, .edgePath.highlight, g.port circle {
+  g.node.highlight rect, .edgePath.highlight, g.port circle {
     stroke: #89C9A2;
   }
-  .highlight.child rect, .edgePath.highlight.child {
+  g.highlight.child rect, .edgePath.highlight.child {
     stroke: #C888B0;
   }
   #edge-labels g.port.highlight {