diff --git a/testsuite/python/stokesian_dynamics.py b/testsuite/python/stokesian_dynamics.py
index 794ec4bf74e..22f2d60fa9c 100644
--- a/testsuite/python/stokesian_dynamics.py
+++ b/testsuite/python/stokesian_dynamics.py
@@ -83,7 +83,7 @@ def falling_spheres(self, time_step, l_factor, t_factor,
             device=self.device, radii={0: 1.0 * l_factor},
             approximation_method=sd_method)
 
-        gravity = constraints.Gravity(
+        gravity = espressomd.constraints.Gravity(
             g=[0, -1.0 * l_factor / (t_factor**2), 0])
         self.system.constraints.add(gravity)
         self.system.time_step = 1.0 * t_factor