diff --git a/desktop_env/envs/__init__.py b/desktop_env/envs/__init__.py
index 870b674..470c203 100644
--- a/desktop_env/envs/__init__.py
+++ b/desktop_env/envs/__init__.py
@@ -12,7 +12,7 @@ import psutil
 import requests
 from tqdm import tqdm
 
-__version__ = "0.1.8"
+__version__ = "0.1.9"
 
 MAX_RETRY_TIMES = 10
 UBUNTU_ARM_URL = "https://huggingface.co/datasets/xlangai/ubuntu_arm/resolve/main/Ubuntu.zip"
@@ -95,6 +95,7 @@ class VirtualMachineManager:
                 file.writelines(new_lines)
 
             # Check and clean on the files inside vms_dir, delete the unregistered ones
+            os.makedirs(vms_dir, exist_ok=True)
             vm_names = os.listdir(vms_dir)
             for vm_name in vm_names:
                 # skip the downloaded .zip file