diff --git a/rps/robotarium_abc.py b/rps/robotarium_abc.py
index bfc7e13..9d4059c 100644
--- a/rps/robotarium_abc.py
+++ b/rps/robotarium_abc.py
@@ -126,6 +126,7 @@ def set_velocities(self, ids: NDArray[np.integer], velocities: NDArray[np.floati
         assert velocities.shape[1] <= self.number_of_robots, (
             f"Column count of velocities ({velocities.shape[1]}) must be <= number_of_robots ({self.number_of_robots})"
         )
+        ids = np.atleast_1d(ids)
         self._velocities[:, ids] = velocities
 
     def robot_colormap(self) -> plt.cm.ScalarMappable: