--- tests/tensor/test_extra_ops.py.orig	2020-12-14 05:42:43.000000000 -0700
+++ tests/tensor/test_extra_ops.py	2021-01-11 11:45:18.625666861 -0700
@@ -845,7 +845,7 @@ class TestUnique(utt.InferShapeTester):
 class TestUniqueAxis(utt.InferShapeTester):
     def setup_method(self):
         super().setup_method()
-        numpy_ver = tuple([int(n) for n in np.__version__.split(".")])
+        numpy_ver = tuple([int(n) for n in np.__version__.split(".")[:2]])
         if numpy_ver >= (1, 13):
             self.expect_success = True
         else: