diff --git a/benchmarks/Schelling/schelling.py b/benchmarks/Schelling/schelling.py
index 47bf521e057..669c1477dd0 100644
--- a/benchmarks/Schelling/schelling.py
+++ b/benchmarks/Schelling/schelling.py
@@ -1,7 +1,7 @@
 """Schelling separation for performance benchmarking."""
 
 from mesa import Model
-from mesa.experimental.cell_space import CellAgent, OrthogonalMooreGrid
+from mesa.spaces import CellAgent, OrthogonalMooreGrid
 from mesa.time import RandomActivation
 
 
diff --git a/benchmarks/WolfSheep/wolf_sheep.py b/benchmarks/WolfSheep/wolf_sheep.py
index 0999fc77842..14863e0db30 100644
--- a/benchmarks/WolfSheep/wolf_sheep.py
+++ b/benchmarks/WolfSheep/wolf_sheep.py
@@ -10,7 +10,7 @@
 import math
 
 from mesa import Model
-from mesa.experimental.cell_space import CellAgent, OrthogonalVonNeumannGrid
+from mesa.spaces import CellAgent, OrthogonalVonNeumannGrid
 from mesa.experimental.devs import ABMSimulator