diff --git a/dace/sdfg/work_depth_analysis/operational_intensity.py b/dace/sdfg/work_depth_analysis/operational_intensity.py
index f9c3836e40..f165b3b510 100644
--- a/dace/sdfg/work_depth_analysis/operational_intensity.py
+++ b/dace/sdfg/work_depth_analysis/operational_intensity.py
@@ -17,9 +17,8 @@
 from dace.transformation.pass_pipeline import FixedPointPipeline
 
 from dace.data import Array
-from dace.sdfg.work_depth_analysis.op_in_helpers import CacheLineTracker, AccessStack
+from dace.sdfg.work_depth_analysis.op_in_helpers import CacheLineTracker, AccessStack, fit_curve, plot, compute_mape
 from dace.sdfg.work_depth_analysis.work_depth import analyze_sdfg, get_tasklet_work
-from dace.sdfg.work_depth_analysis.extrapolation import fit_curve, plot, compute_mape
 
 
 class SymbolRange():
diff --git a/tests/sdfg/work_depth_tests.py b/tests/sdfg/work_depth_test.py
similarity index 100%
rename from tests/sdfg/work_depth_tests.py
rename to tests/sdfg/work_depth_test.py