diff --git a/superset/datasets/commands/importers/v1/utils.py b/superset/datasets/commands/importers/v1/utils.py
index 5e59545ae78b..ad09235a26d3 100644
--- a/superset/datasets/commands/importers/v1/utils.py
+++ b/superset/datasets/commands/importers/v1/utils.py
@@ -30,7 +30,7 @@
 
 from superset.connectors.sqla.models import SqlaTable
 from superset.models.core import Database
-from superset.utils.core import get_example_database, get_main_database
+from superset.utils.core import get_example_database
 
 logger = logging.getLogger(__name__)
 
@@ -136,7 +136,7 @@ def load_data(
             df[column_name] = pd.to_datetime(df[column_name])
 
     # reuse session when loading data if possible, to make import atomic
-    if example_database.sqlalchemy_uri == get_main_database().sqlalchemy_uri:
+    if example_database.sqlalchemy_uri == get_example_database().sqlalchemy_uri:
         logger.info("Loading data inside the import transaction")
         connection = session.connection()
     else: