diff --git a/python/pycute/swizzle.py b/python/pycute/swizzle.py
index 3f73d1bc..308aee0c 100644
--- a/python/pycute/swizzle.py
+++ b/python/pycute/swizzle.py
@@ -75,7 +75,7 @@ class Swizzle:
 
   # Size of the domain
   def size(self):
-    return 1 << (bits + base + abs(shift))
+    return 1 << (self.bits + self.base + abs(self.shift))
 
   # Size of the codomain
   def cosize(self):