diff --git a/tests/performance/_constants.py b/tests/performance/_constants.py
index f114754..98c2754 100644
--- a/tests/performance/_constants.py
+++ b/tests/performance/_constants.py
@@ -14,9 +14,9 @@
 }
 
 GRAPH_SETUP_CODES = {
-    "rustworkx": "from power_grid_model_ds.model.grids.base import Grid;"
-    + "from power_grid_model_ds.data_source.generator.grid_generators import RadialGridGenerator;"
-    + "from power_grid_model_ds.model.graphs.models import RustworkxGraphModel;"
+    "rustworkx": "from power_grid_model_ds import Grid;"
+    + "from power_grid_model_ds.generators import RadialGridGenerator;"
+    + "from power_grid_model_ds.graph_models import RustworkxGraphModel;"
     + "grid=RadialGridGenerator(nr_nodes={graph_size}, grid_class=Grid, graph_model=RustworkxGraphModel).run()",
 }
 
diff --git a/tests/performance/graph_performance_tests.py b/tests/performance/graph_performance_tests.py
index 8b242df..50b1aae 100644
--- a/tests/performance/graph_performance_tests.py
+++ b/tests/performance/graph_performance_tests.py
@@ -48,9 +48,9 @@ def perftest_add_node():
 
 
 if __name__ == "__main__":
-    perftest_initialize()
-    perftest_set_feeder_ids()
+    # perftest_initialize()
+    # perftest_set_feeder_ids()
     perftest_get_components()
-    perftest_delete_node()
-    perftest_add_node()
-    perftest_from_arrays()
+    # perftest_delete_node()
+    # perftest_add_node()
+    # perftest_from_arrays()