diff --git a/scripts/irods/upgrade_configuration.py b/scripts/irods/upgrade_configuration.py
index e210547b04..22ee972a10 100644
--- a/scripts/irods/upgrade_configuration.py
+++ b/scripts/irods/upgrade_configuration.py
@@ -11,6 +11,7 @@
 from . import lib
 from . import paths
 from .exceptions import IrodsError, IrodsWarning
+from .configuration import IrodsConfig
 
 def schema_version_as_int(schema_version):
     if isinstance(schema_version, str) and schema_version[0] == 'v':