diff --git a/tests/test_agv_robots.py b/tests/test_agv_robots.py
index 97b30b1..a7da8a7 100644
--- a/tests/test_agv_robots.py
+++ b/tests/test_agv_robots.py
@@ -91,15 +91,15 @@ def tearDown(self) -> None:
 
     def test_calculate_wheel_commands(self):
 
-        wheel_commands = self.robot._calculate_wheel_comands(1, 0)
+        wheel_commands = self.robot._calculate_wheel_comands([1, 0])
         self.assertTrue(np.allclose(wheel_commands, [10, 10]))
 
 
-        wheel_commands = self.robot._calculate_wheel_comands(0, 1)
+        wheel_commands = self.robot._calculate_wheel_comands([0, 1])
         self.assertTrue(np.allclose(wheel_commands, [-2.5, 2.5]))
 
 
-        wheel_commands = self.robot._calculate_wheel_comands(1, 1)
+        wheel_commands = self.robot._calculate_wheel_comands([1, 1])
         self.assertTrue(np.allclose(wheel_commands, [7.5, 12.5]))
 
     def test_set_velocity(self):