diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
index eadd0a3c464f..b6be05a3d69e 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -446,11 +446,9 @@ def slice(self, inputs, input_types):
         stride = inputs[4]
 
         target_begin, is_begin_const = try_infer_value(
-            inputs[2], lambda ret: ret.astype(np.int).item(0)
-        )
-        target_end, is_end_const = try_infer_value(
-            inputs[3], lambda ret: ret.astype(np.int).item(0)
+            inputs[2], lambda ret: ret.astype(int).item(0)
         )
+        target_end, is_end_const = try_infer_value(inputs[3], lambda ret: ret.astype(int).item(0))
 
         # A fast path when slicing is nop.
         if (