diff --git a/tests/test_basis.py b/tests/test_basis.py
index 9f5d3e4..dab2f41 100644
--- a/tests/test_basis.py
+++ b/tests/test_basis.py
@@ -26,6 +26,7 @@
 
 import numpy as np
 import pytest
+from opt_einsum import contract
 from sparse import COO
 
 import filter_functions as ff
@@ -142,7 +143,7 @@ def test_basis_properties(self):
 
             if base.d < 8:
                 # Test very resource intense
-                ref = np.einsum('iab,jbc,kcd,lda', *(base,)*4)
+                ref = contract('iab,jbc,kcd,lda', *(base,)*4)
                 self.assertArrayAlmostEqual(base.four_element_traces.todense(), ref, atol=1e-16)
 
                 # Test setter