diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 391bc61..cf118b5 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -492,7 +492,7 @@
         "            last_output = log_line\n",
         "\n",
         "# Finish the training process\n",
-        "process.wait()"
+        "process.wait()",
         "if Disconnect_after_training :",
         "  time.sleep(40)",
         "  runtime.unassign()"