diff --git a/decensor.py b/decensor.py
index f4992ec..32e7344 100644
--- a/decensor.py
+++ b/decensor.py
@@ -75,8 +75,8 @@ def get_mask(x_batch):
         raw = x_batch[i]
         raw = np.array((raw + 1) * 127.5, dtype=np.uint8)
         m = np.zeros((IMAGE_SIZE, IMAGE_SIZE, 1), dtype=np.uint8)
-        for x in xrange(IMAGE_SIZE):
-            for y in xrange(IMAGE_SIZE):
+        for x in range(IMAGE_SIZE):
+            for y in range(IMAGE_SIZE):
                 if np.array_equal(raw[x][y], [0, 255, 0]):
                     m[x, y] = 1
         mask.append(m)