diff --git a/gluoncv/torch/data/video_cls/multigrid_helper.py b/gluoncv/torch/data/video_cls/multigrid_helper.py
index 05fc2533bf..127c2b5d50 100644
--- a/gluoncv/torch/data/video_cls/multigrid_helper.py
+++ b/gluoncv/torch/data/video_cls/multigrid_helper.py
@@ -3,8 +3,8 @@
 import numpy as np
 
 from torch.utils.data import Sampler
-from torch._six import int_classes as _int_classes
 
+_int_classes = int
 
 __all__ = ['multiGridHelper', 'MultiGridBatchSampler']