diff --git a/autoaugment.py b/autoaugment.py
index 339734c..04942f3 100644
--- a/autoaugment.py
+++ b/autoaugment.py
@@ -173,7 +173,7 @@ def __init__(self, p1, operation1, magnitude_idx1, p2, operation2, magnitude_idx
             "translateY": np.linspace(0, 150 / 331, 10),
             "rotate": np.linspace(0, 30, 10),
             "color": np.linspace(0.0, 0.9, 10),
-            "posterize": np.round(np.linspace(8, 4, 10), 0).astype(np.int),
+            "posterize": np.round(np.linspace(8, 4, 10), 0).astype(np.int32),
             "solarize": np.linspace(256, 0, 10),
             "contrast": np.linspace(0.0, 0.9, 10),
             "sharpness": np.linspace(0.0, 0.9, 10),