diff --git a/python/mujoco/specs_test.py b/python/mujoco/specs_test.py
index f8ba454e..d2f07eca 100644
--- a/python/mujoco/specs_test.py
+++ b/python/mujoco/specs_test.py
@@ -1323,7 +1323,7 @@ class SpecsTest(absltest.TestCase):
   def test_actuator_shortname(self):
     spec = mujoco.MjSpec()
     actuator = spec.add_actuator(
-        gainprm=np.zeros((10, 1)),
+        gainprm=np.zeros((10,)),
         dyntype=mujoco.mjtDyn.mjDYN_FILTER,
         gaintype=mujoco.mjtGain.mjGAIN_AFFINE,
         biastype=mujoco.mjtBias.mjBIAS_AFFINE,