diff --git a/scripts/run_uav_visloc.py b/scripts/run_uav_visloc.py
index e7329ca..ebccaa8 100644
--- a/scripts/run_uav_visloc.py
+++ b/scripts/run_uav_visloc.py
@@ -73,7 +73,11 @@ def main() -> None:
     )
 
     input_conf = InputConfig(image_size=256)
-    seg_conf = SegConfig()
+    seg_conf = SegConfig(prompts=[
+        "background", "building", "road", "vegetation", "water",
+        "sand and gravel ground", "rocky terrain", "farmland",
+        "railway", "parking lot", "sidewalk",
+    ])
     models_conf = ModelsConfig()
 
     setup_logging(