diff --git a/desktop_env/envs/desktop_env.py b/desktop_env/envs/desktop_env.py
index ed2f8c4..fbfb739 100644
--- a/desktop_env/envs/desktop_env.py
+++ b/desktop_env/envs/desktop_env.py
@@ -137,7 +137,7 @@ class DesktopEnv(gym.Env):
         logger.info("Getting IP Address...")
         for _ in range(max_retries):
             try:
-                output = _execute_command(["vmrun", "-T", "ws", "getGuestIPAddress", self.path_to_vm]).strip()
+                output = _execute_command(["vmrun", "-T", "ws", "getGuestIPAddress", self.path_to_vm, "-wait"]).strip()
                 logger.info(f"IP address: {output}")
                 return output
             except Exception as e: