diff --git a/tests/test_ntrfc_pyvistautils.py b/tests/test_ntrfc_pyvistautils.py
index 18159f2eb91c8cf2ea45b79c1576987b1e83ccf7..203665f9da289fb4782940a009c3326fa78fdcad 100644
--- a/tests/test_ntrfc_pyvistautils.py
+++ b/tests/test_ntrfc_pyvistautils.py
@@ -4,7 +4,7 @@
 def test_massflow_plane():
     import numpy as np
     import pyvista as pv
-    from ntrfc.pyvista_utils.surface import massflow_plane
+    from ntrfc.utils.pyvista_utils.surface import massflow_plane
 
     plane = pv.Plane()
     numcells = plane.number_of_cells
@@ -19,7 +19,7 @@ def test_massflow_plane():
 def test_areaave_plane():
     import numpy as np
     import pyvista as pv
-    from ntrfc.pyvista_utils.surface import areaave_plane
+    from ntrfc.utils.pyvista_utils.surface import areaave_plane
 
     plane = pv.Plane()
     plane["U"] = np.ones(plane.number_of_cells)