diff --git a/embodichain/lab/sim/utility/sim_utils.py b/embodichain/lab/sim/utility/sim_utils.py
index d708aebc..523c7fef 100644
--- a/embodichain/lab/sim/utility/sim_utils.py
+++ b/embodichain/lab/sim/utility/sim_utils.py
@@ -100,7 +100,7 @@ def get_drive_type(drive_pros):
     elif drive_type == "acceleration":
         drive_type = DriveType.ACCELERATION
     elif drive_type == "none":
-        return DriveType.NONE
+        drive_type = DriveType.NONE
     else:
         logger.log_error(f"Unknow drive type {drive_type}")