diff --git a/gym_pybullet_drones/envs/BetaAviary.py b/gym_pybullet_drones/envs/BetaAviary.py
index e4d37a2e4..c162681b8 100644
--- a/gym_pybullet_drones/envs/BetaAviary.py
+++ b/gym_pybullet_drones/envs/BetaAviary.py
@@ -195,7 +195,8 @@ def _actionSpace(self):
         Returns
         -------
         spaces.Box
-            An ndarray of shape (NUM_DRONES, 4) for the commanded RPMs.
+            An ndarray of shape (NUM_DRONES, 4) for the commanded RPMs. 
+            correct as in #196
 
         """
         #### Action vector ######## P0            P1            P2            P3