diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_import_error.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_import_error.py
index 173e57592e49f..2b7c72e864646 100644
--- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_import_error.py
+++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_import_error.py
@@ -27,6 +27,7 @@
 from airflow.models.errors import ParseImportError
 from airflow.utils.session import NEW_SESSION, provide_session
 
+from tests_common.test_utils.asserts import assert_queries_count
 from tests_common.test_utils.db import clear_db_dag_bundles, clear_db_dags, clear_db_import_errors
 from tests_common.test_utils.format_datetime import from_datetime_to_zulu_without_ms
 
@@ -322,7 +323,8 @@ def test_get_import_errors(
         expected_total_entries,
         expected_filenames,
     ):
-        response = test_client.get("/importErrors", params=query_params)
+        with assert_queries_count(2):
+            response = test_client.get("/importErrors", params=query_params)
 
         assert response.status_code == expected_status_code
         if expected_status_code != 200:
@@ -372,7 +374,8 @@ def test_user_can_not_read_all_dags_in_file(
             mock_get_auth_manager, batch_is_authorized_dag_return_value
         )
         # Act
-        response = test_client.get("/importErrors")
+        with assert_queries_count(3):
+            response = test_client.get("/importErrors")
         # Assert
         mock_get_authorized_dag_ids.assert_called_once_with(method="GET", user=mock.ANY)
         assert response.status_code == 200