diff --git a/setup.py b/setup.py
index 665d5028..a9acd35c 100644
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,7 @@
     "jax[cpu]",
     "numpy",
     "pandas",
+    "matplotlib",
 ]
 
 
diff --git a/tests/test_swc.py b/tests/test_swc.py
index e4f9d236..eb3d8adf 100644
--- a/tests/test_swc.py
+++ b/tests/test_swc.py
@@ -20,7 +20,7 @@
 def test_swc_reader_lengths():
     fname = "morph.swc"
 
-    _, pathlengths, _, _ = nx.utils.read_swc(fname, max_branch_len=2000.0)
+    _, pathlengths, _, _ = nx.utils.swc.swc_to_neurax(fname, max_branch_len=2000.0)
     pathlengths = np.asarray(pathlengths)[1:]
 
     for sec in h.allsec():
@@ -58,7 +58,7 @@ def test_swc_radius():
     # Can not use full morphology because of branch sorting.
     fname = "morph_250.swc"
 
-    _, pathlen, radius_fns, _ = nx.utils.read_swc(
+    _, pathlen, radius_fns, _ = nx.utils.swc.swc_to_neurax(
         fname, max_branch_len=2000.0, sort=False
     )
     neurax_diams = []