diff --git a/webui_train.py b/webui_train.py
index 2b749b1..fb894cd 100644
--- a/webui_train.py
+++ b/webui_train.py
@@ -20,7 +20,8 @@ def get_path(model_name):
         dataset_path = os.path.join(
             "/content/drive/MyDrive/Style-Bert-VITS2/Data", model_name
         )
-    dataset_path = os.path.join("Data", model_name)
+    else:
+        dataset_path = os.path.join("Data", model_name)
     lbl_path = os.path.join(dataset_path, "esd.list")
     train_path = os.path.join(dataset_path, "train.list")
     val_path = os.path.join(dataset_path, "val.list")