diff --git a/scripts/run_uav_visloc.py b/scripts/run_uav_visloc.py
index d7ab5c0..a5f5a08 100644
--- a/scripts/run_uav_visloc.py
+++ b/scripts/run_uav_visloc.py
@@ -73,7 +73,7 @@ def main() -> None:
     )
 
     input_conf = InputConfig(image_size=256)
-    seg_conf = SegConfig(prompts=[
+    seg_conf = SegConfig(threshold=0.1, prompts=[
         "background",                   # 0
         "building",                     # 1
         "road",                         # 2