diff --git a/numba_rvsdg/tests/test_ast_transforms.py b/numba_rvsdg/tests/test_ast_transforms.py
index bfc7dc08..d58c60e5 100644
--- a/numba_rvsdg/tests/test_ast_transforms.py
+++ b/numba_rvsdg/tests/test_ast_transforms.py
@@ -31,9 +31,9 @@ def compare(
         self,
         function: Callable[..., Any],
         expected: dict[str, dict[str, Any]],
-        unreachable: set[int] = set(),
-        empty: set[int] = set(),
-        arguments: list[any] = [],
+        unreachable: set[str] = set(),
+        empty: set[str] = set(),
+        arguments: list[Any] = [],
     ):
         # Execute function with first argument, if given. Ensure function is
         # sane and make sure it's picked up by coverage.