diff --git a/tests/test_init.py b/tests/test_init.py
index 72a490a..0ec96bd 100644
--- a/tests/test_init.py
+++ b/tests/test_init.py
@@ -1,5 +1,4 @@
-from pet_neighbors_convert import load_neighbors_convert
-
-
 def test_init():
-    load_neighbors_convert()
+    import torch
+    import pet_neighbors_convert
+    torch.ops.neighbors_convert.process
diff --git a/tox.ini b/tox.ini
index a665ddb..f2e23a8 100644
--- a/tox.ini
+++ b/tox.ini
@@ -8,7 +8,6 @@ passenv = *
 
 [testenv:build]
 description = Asserts package build integrity.
-usedevelop = true
 deps =
     build
     check-manifest
@@ -21,6 +20,5 @@ commands =
 
 [testenv:tests]
 description = Run test suite with pytest and {basepython}.
-usedevelop = true
 deps = pytest
 commands = pytest {posargs}