diff --git a/test/test_viz.py b/test/test_viz.py
index 8296edfa8a..527c20a7cd 100644
--- a/test/test_viz.py
+++ b/test/test_viz.py
@@ -4,7 +4,7 @@ import itertools
 from tinygrad import Tensor, dtypes
 from tinygrad.helpers import Context, getenv
 from tinygrad.engine.realize import lower_schedule
-from viz.serve import GraphRewriteMetadata, get_metadata, _uop_to_json
+from tinygrad.viz.serve import GraphRewriteMetadata, get_metadata, _uop_to_json
 from tinygrad.ops import TRACK_MATCH_STATS, TrackedPatternMatcher, UPat, UOps, UOp, graph_rewrite, contexts, track_rewrites
 
 def group_rewrites(kernels:List[GraphRewriteMetadata]): return {k:list(v) for k,v in itertools.groupby(kernels, lambda x:x.loc)}
diff --git a/tinygrad/ops.py b/tinygrad/ops.py
index 1b5864f7e5..12ef986968 100644
--- a/tinygrad/ops.py
+++ b/tinygrad/ops.py
@@ -638,7 +638,7 @@ if TRACK_MATCH_STATS:
         pickle.dump(contexts, f)
     if getenv("VIZ"):
       os.environ["VIZ"] = "0"
-      os.execv(sys.executable, [sys.executable] + [os.path.join(os.path.dirname(__file__), "..", "viz", "serve.py")])
+      os.execv(sys.executable, [sys.executable] + [os.path.join(os.path.dirname(__file__), ".", "viz", "serve.py")])
     if getenv("PRINT_MATCH_STATS", 1):
       ret = [0,0,0.0,0.0]
       for k,v in sorted(list(match_stats.items()), key=lambda x: x[1][2]):
diff --git a/viz/README b/tinygrad/viz/README
similarity index 100%
rename from viz/README
rename to tinygrad/viz/README
diff --git a/viz/favicon.svg b/tinygrad/viz/favicon.svg
similarity index 100%
rename from viz/favicon.svg
rename to tinygrad/viz/favicon.svg
diff --git a/viz/index.html b/tinygrad/viz/index.html
similarity index 100%
rename from viz/index.html
rename to tinygrad/viz/index.html
diff --git a/viz/serve.py b/tinygrad/viz/serve.py
similarity index 100%
rename from viz/serve.py
rename to tinygrad/viz/serve.py