diff --git a/nlstruct/torch_utils.py b/nlstruct/torch_utils.py
index 1572c55..fe6cf54 100644
--- a/nlstruct/torch_utils.py
+++ b/nlstruct/torch_utils.py
@@ -2,7 +2,11 @@
 import random
 import re
 import math
-from collections import defaultdict, Sequence
+try:
+    from collections.abc import Sequence
+except ImportError:
+    from collections import Sequence
+from collections import defaultdict
 from collections import namedtuple
 from contextlib import contextmanager
 from itertools import zip_longest