diff --git a/src/pybricks/robotics.pyi b/src/pybricks/robotics.pyi
index ea29fce..aec5a42 100644
--- a/src/pybricks/robotics.pyi
+++ b/src/pybricks/robotics.pyi
@@ -14,7 +14,7 @@ class DriveBase:
         wheel_diameter: Union[int, float],
         axle_track: Union[int, float],
     ): ...
-    def drive(self, drive_speed: int, turn_rate: int) -> None: ...
+    def drive(self, speed: int, turn_rate: int) -> None: ...
     def stop(self) -> None: ...
     def distance(self) -> int: ...
     def angle(self) -> int: ...