diff --git a/py-polars/tests/unit/operations/unique/test_unique.py b/py-polars/tests/unit/operations/unique/test_unique.py
index d6b417fa7098..5034c8fab87d 100644
--- a/py-polars/tests/unit/operations/unique/test_unique.py
+++ b/py-polars/tests/unit/operations/unique/test_unique.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from datetime import date
+from typing import TYPE_CHECKING, Any
 
 import pytest
 
@@ -8,6 +9,9 @@
 from polars.exceptions import ColumnNotFoundError
 from polars.testing import assert_frame_equal, assert_series_equal
 
+if TYPE_CHECKING:
+    from polars._typing import PolarsDataType
+
 
 def test_unique_predicate_pd() -> None:
     lf = pl.LazyFrame(
@@ -175,7 +179,9 @@ def test_unique_with_null() -> None:
     ],
 )
 def test_unique_with_bad_subset(
-    input_json_data: dict, input_schema: dict, subset: str | list[str]
+    input_json_data: dict[str, list[Any]],
+    input_schema: dict[str, PolarsDataType],
+    subset: str | list[str],
 ) -> None:
     df = pl.DataFrame(input_json_data, schema=input_schema)