diff --git a/notebook.ipynb b/notebook.ipynb
index ca0581f..d41bc59 100755
--- a/notebook.ipynb
+++ b/notebook.ipynb
@@ -42,7 +42,7 @@
             "%cd ai-voice-cloning\n",
             "\n",
             "!git submodule init\n",
-            "!git submodule update\n",
+            "!git submodule update --remote\n",
             "\n",
             "# TODO: fix venvs working for subprocess.Popen calling a bash script\n",
             "#!apt install python3.8-venv\n",
@@ -107,7 +107,7 @@
             "!mkdir /content/drive/MyDrive/ai-voice-cloning/\n",
             "!mv /content/drive/MyDrive/{training,results,voices,config} /content/drive/MyDrive/ai-voice-cloning\n",
             "!mkdir /content/drive/MyDrive/ai-voice-cloning/{training,results,voices,config}\n",
-            "!ln -s /content/drive/MyDrive/ai-voice-cloning/{training,results,voices,config} /content/drive/MyDrive/ai-voice-cloning/"
+            "!ln -s /content/drive/MyDrive/ai-voice-cloning/{training,results,voices,config} ./"
          ],
          "metadata":{
             "id":"SGt9gyvubveT"
diff --git a/src/webui.py b/src/webui.py
index a2e306e..d6c16ef 100755
--- a/src/webui.py
+++ b/src/webui.py
@@ -205,7 +205,8 @@ def import_training_settings_proxy( epochs, learning_rate, learning_rate_schedul
 	outdir = f'./training/{voice}-finetune/'
 
 	in_config_path = f"{indir}/train.yaml"
-	out_configs = sorted([d[:-5] for d in os.listdir(outdir) if d[-5:] == ".yaml" ])
+	if os.path.isdir(outdir):
+		out_configs = sorted([d[:-5] for d in os.listdir(outdir) if d[-5:] == ".yaml" ])
 	if len(out_configs) > 0:
 		out_config_path = f'{outdir}/{out_configs[-1]}.yaml'