diff --git a/tinygrad/schedule/rangeify2.py b/tinygrad/schedule/rangeify2.py
index 36be48e6d3..2582844fc8 100644
--- a/tinygrad/schedule/rangeify2.py
+++ b/tinygrad/schedule/rangeify2.py
@@ -154,7 +154,7 @@ class SplitCtx:
   range_number:int = -1
 
 def _split_graph(ctx:SplitCtx, u:UOp) -> UOp:
-  assert len(u.shape) == 1, "rangeify needs to reduce to a single idx"
+  assert len(u.shape) <= 1, f"rangeify needs to reduce to a single idx, not {u.shape}"
   ctx.call_args.append(u)
   return u.param_like(len(ctx.call_args)-1)