diff --git a/discO/plugin/galpy/tests/test_sample.py b/discO/plugin/galpy/tests/test_sample.py
index 030efe25..ee460e0a 100644
--- a/discO/plugin/galpy/tests/test_sample.py
+++ b/discO/plugin/galpy/tests/test_sample.py
@@ -42,7 +42,7 @@ def setup_class(cls):
         # make potential
         cls.mass = 1e12 * u.solMass
 
-        hernquist_pot = HernquistPotential(amp=2 * cls.mass, ro=8*u.kpc, vo=220*u.km/u.s)
+        hernquist_pot = HernquistPotential(amp=2 * cls.mass, ro=8 * u.kpc, vo=220 * u.km / u.s)
         hernquist_pot.turn_physical_on(ro=8 * u.kpc, vo=220 * u.km / u.s)
         cls.potential = hernquist_pot
 
@@ -155,7 +155,7 @@ def setup_class(cls):
         # make potential
         cls.mass = 1e12 * u.solMass
 
-        hernquist_pot = HernquistPotential(amp=2 * cls.mass, ro=8*u.kpc, vo=220*u.km/u.s)
+        hernquist_pot = HernquistPotential(amp=2 * cls.mass, ro=8 * u.kpc, vo=220 * u.km / u.s)
         hernquist_pot.turn_physical_on(ro=8 * u.kpc, vo=220 * u.km / u.s)
         cls.potential = hernquist_pot