diff --git a/supervisely/serve/config.json b/supervisely/serve/config.json
index 2d9948e25bed..a14eff45518d 100644
--- a/supervisely/serve/config.json
+++ b/supervisely/serve/config.json
@@ -11,7 +11,7 @@
     "serve"
   ], 
   "description": "Deploy model as REST API service",
-  "docker_image": "supervisely/yolov5:1.0.0",
+  "docker_image": "supervisely/yolov5:1.0.1",
   "instance_version": "6.6.7",
   "entrypoint": "python -m uvicorn main:m.app --app-dir ./supervisely/serve/src --host 0.0.0.0 --port 8000 --ws websockets",
   "port": 8000,
diff --git a/supervisely/train/config.json b/supervisely/train/config.json
index ae12c2c086fc..82bb71026aaa 100644
--- a/supervisely/train/config.json
+++ b/supervisely/train/config.json
@@ -10,7 +10,7 @@
     "train"
   ],
   "description": "Dashboard to configure and monitor training",
-  "docker_image": "supervisely/yolov5:1.0.0",
+  "docker_image": "supervisely/yolov5:1.0.1",
   "instance_version": "6.4.57",
   "main_script": "supervisely/train/src/sly_train.py",
   "gui_template": "supervisely/train/src/gui.html",
diff --git a/supervisely/train/src/sly_train_globals.py b/supervisely/train/src/sly_train_globals.py
index 5b2e47f31d4a..5c6f083fe3fe 100644
--- a/supervisely/train/src/sly_train_globals.py
+++ b/supervisely/train/src/sly_train_globals.py
@@ -18,6 +18,11 @@
 sys.path.append(ui_sources_dir)
 sly.logger.info(f"Added to sys.path: {ui_sources_dir}")
 
+# currently PYTHONPATH is ":/app/repo"
+python_path = os.environ.get("PYTHONPATH", "/app/repo").replace(":", "")
+sys.path.insert(0, python_path)
+
+
 load_dotenv(os.path.join(root_source_dir, "supervisely", "train", "debug.env"))
 load_dotenv(os.path.join(root_source_dir, "supervisely", "train", "secret_debug.env"), override=True)