diff --git a/tests/sql_parse_tests.py b/tests/sql_parse_tests.py
index d85271d50..1722d6305 100644
--- a/tests/sql_parse_tests.py
+++ b/tests/sql_parse_tests.py
@@ -540,7 +540,7 @@ class SupersetTestCase(unittest.TestCase):
         SELECT * FROM birth_names;
         -- some comment
         """
-        parsed = sql_parse.ParsedQuery(multi_sql)
+        parsed = ParsedQuery(multi_sql)
         statements = parsed.get_statements()
         self.assertEqual(len(statements), 1)
         expected = ["SELECT * FROM birth_names"]