diff --git a/scripts/geodata/osm/osm_address_training_data.py b/scripts/geodata/osm/osm_address_training_data.py
index e4588e99..0dfca25c 100644
--- a/scripts/geodata/osm/osm_address_training_data.py
+++ b/scripts/geodata/osm/osm_address_training_data.py
@@ -51,6 +51,7 @@ sys.path.append(os.path.realpath(os.path.join(os.pardir, os.pardir)))
 
 from geodata.address_expansions.abbreviations import abbreviate
 from geodata.address_expansions.gazetteers import *
+from geodata.addresses.components import AddressComponents
 from geodata.coordinates.conversion import *
 from geodata.geonames.db import GeoNamesDB
 from geodata.language_id.disambiguation import *