diff --git a/tests/vegalite/v5/test_api.py b/tests/vegalite/v5/test_api.py
index 5c25e0a80..a0de4b36d 100644
--- a/tests/vegalite/v5/test_api.py
+++ b/tests/vegalite/v5/test_api.py
@@ -254,7 +254,7 @@ def test_chart_operations():
 
 def test_when() -> None:
     select = alt.selection_point(name="select", on="click")
-    condition = alt.condition(select, alt.value(1), "two", empty=False).get("condition")
+    condition = alt.condition(select, alt.value(1), "two", empty=False)["condition"]
     condition.pop("value")
     when = alt.when(select, empty=False)
     when_constraint = alt.when(Origin="Europe")