diff --git a/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py b/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py
index c333400ba62d8..7f87040a5da61 100644
--- a/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py
+++ b/superset/migrations/versions/2024-05-24_11-31_02f4f7811799_remove_sl__tables.py
@@ -31,13 +31,13 @@
 
 
 def upgrade():
-    op.drop_table("sl_columns")
+    op.drop_table("sl_dataset_columns")
     op.drop_table("sl_table_columns")
     op.drop_table("sl_dataset_tables")
-    op.drop_table("sl_dataset_columns")
+    op.drop_table("sl_columns")
     op.drop_table("sl_tables")
-    op.drop_table("sl_datasets")
     op.drop_table("sl_dataset_users")
+    op.drop_table("sl_datasets")
 
 
 def downgrade():
diff --git a/tests/integration_tests/datasets/api_tests.py b/tests/integration_tests/datasets/api_tests.py
index 6cc3cc8828dc0..5cabe3c585015 100644
--- a/tests/integration_tests/datasets/api_tests.py
+++ b/tests/integration_tests/datasets/api_tests.py
@@ -38,7 +38,6 @@
     DAODeleteFailedError,
     DAOUpdateFailedError,
 )
-from superset.datasets.models import Dataset  # noqa: F401
 from superset.extensions import db, security_manager
 from superset.models.core import Database
 from superset.models.slice import Slice