diff --git a/scripts/geodata/osm/osm_address_training_data.py b/scripts/geodata/osm/osm_address_training_data.py
index 06eb360c..cb05dfda 100644
--- a/scripts/geodata/osm/osm_address_training_data.py
+++ b/scripts/geodata/osm/osm_address_training_data.py
@@ -924,8 +924,8 @@ if __name__ == '__main__':
 
     language_rtree = LanguagePolygonIndex.load(args.language_rtree_dir)
     rtree = None
-    if args.osm_rtree_dir:
-        osm_rtree = OSMReverseGeocoder.load(args.osm_rtree_dir)
+    if args.rtree_dir:
+        osm_rtree = OSMReverseGeocoder.load(args.rtree_dir)
 
     if args.quattroshapes_rtree_dir:
         quattroshapes_rtree = QuattroshapesReverseGeocoder.load(args.quattroshapes_rtree_dir)