diff --git a/scripts/run_uav_visloc.py b/scripts/run_uav_visloc.py
index ebccaa8..cb4ab81 100644
--- a/scripts/run_uav_visloc.py
+++ b/scripts/run_uav_visloc.py
@@ -78,7 +78,7 @@ def main() -> None:
         "sand and gravel ground", "rocky terrain", "farmland",
         "railway", "parking lot", "sidewalk",
     ])
-    models_conf = ModelsConfig()
+    models_conf = ModelsConfig(weights_dir=str(_PROJECT_ROOT / "in" / "weights"))
 
     setup_logging(
         pipeline_conf.log_level,