fix last test

PostHog · Sep 5, 2024 · 49d150c · 49d150c
1 parent ccae49f
commit 49d150c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/posthog/hogql_queries/insights/funnels/test/test_funnel_unordered.py b/posthog/hogql_queries/insights/funnels/test/test_funnel_unordered.py
@@ -1,7 +1,6 @@
 from datetime import datetime
 from typing import cast
 
-from rest_framework.exceptions import ValidationError
 
 from posthog.constants import INSIGHT_FUNNELS, FunnelOrderType
 from posthog.hogql.errors import ExposedHogQLError
@@ -1132,7 +1131,7 @@ def test_funnel_exclusions_invalid_params(self):
         }
 
         query = cast(FunnelsQuery, filter_to_query(filters))
-        self.assertRaises(ValidationError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
+        self.assertRaises(ExposedHogQLError, lambda: FunnelsQueryRunner(query=query, team=self.team).calculate())
 
         # partial windows not allowed for unordered
         filters = {