diff --git a/timesformer/models/vit_utils.py b/timesformer/models/vit_utils.py
index 9ce6a93..8c799c6 100755
--- a/timesformer/models/vit_utils.py
+++ b/timesformer/models/vit_utils.py
@@ -11,7 +11,7 @@
 from timesformer.models.helpers import load_pretrained
 from .build import MODEL_REGISTRY
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc as container_abcs
 
 DEFAULT_CROP_PCT = 0.875
 IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)