diff --git a/tests/test_from_nir.py b/tests/test_from_nir.py
index d6d09ae..3c07d12 100644
--- a/tests/test_from_nir.py
+++ b/tests/test_from_nir.py
@@ -29,9 +29,9 @@ def test_extract_empty():
 
 
 def test_extract_illegal_name():
-    graph = nir.NIRGraph({"a.b": nir.Input(np.ones(1))}, [])
+    graph = nir.NIRGraph({"a.b": nir.Input(np.ones(1)), "a.c": nir.Linear(np.array([[1.]]))}, [("a.b", "a.c")])
     torch_graph = load(graph, _torch_model_map)
-    assert "a_b" in torch_graph._modules
+    assert "a_c" in torch_graph._modules
 
 
 def test_extract_lin():
diff --git a/tests/test_requirements.txt b/tests/test_requirements.txt
index 351614b..50339f8 100644
--- a/tests/test_requirements.txt
+++ b/tests/test_requirements.txt
@@ -3,4 +3,4 @@ torch
 nir
 ruff
 git+https://github.com/norse/norse
-sinabs
\ No newline at end of file
+git+https://github.com/synsense/sinabs@dev/nir