diff --git a/tests/unit_tests/datasets/commands/export_test.py b/tests/unit_tests/datasets/commands/export_test.py
index 0dc01ed3e92a..3d726a6fe63e 100644
--- a/tests/unit_tests/datasets/commands/export_test.py
+++ b/tests/unit_tests/datasets/commands/export_test.py
@@ -18,7 +18,7 @@
 
 from sqlalchemy.orm.session import Session
 
-from superset import db
+from superset.extensions import db
 from superset.utils import json