diff --git a/tests/expr_and_series/n_unique_test.py b/tests/expr_and_series/n_unique_test.py index 90bffb04bc..1bcbe89fdc 100644 --- a/tests/expr_and_series/n_unique_test.py +++ b/tests/expr_and_series/n_unique_test.py @@ -13,7 +13,7 @@ def test_n_unique(constructor: Constructor) -> None: df = nw.from_native(constructor(data)) - result = df.select(nw.all().n_unique()) + result = df.select(nw.col("a", "b").n_unique()) expected = {"a": [3], "b": [4]} assert_equal_data(result, expected)