diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py
index a0987e2..72db0a8 100644
--- a/more_math/Parser/UnifiedMathVisitor.py
+++ b/more_math/Parser/UnifiedMathVisitor.py
@@ -2957,9 +2957,9 @@ class UnifiedMathVisitor(MathExprVisitor):
         shape_val = yield ctx.indexExpr()
 
         if self._is_list(shape_val):
-            shape = self._to_int(v, ctx, "tensor")
+            shape = self._to_int(shape_val, ctx, "tensor")
         elif self._is_tensor(shape_val):
-            shape = self._to_int(v, ctx, "tensor").flatten().tolist()
+            shape = self._to_int(shape_val, ctx, "tensor").flatten().tolist()
         else:
             shape = [self._to_int(shape_val, ctx, "tensor")]