diff --git a/tests/import_test.py b/tests/import_test.py
index 18be0c2..d3573e7 100644
--- a/tests/import_test.py
+++ b/tests/import_test.py
@@ -1,2 +1,3 @@
 def test_import() -> None:
-    import parity_tensor
+    from parity_tensor import ParityTensor
+    assert isinstance(ParityTensor, type)