diff --git a/tests/test_shaker.py b/tests/test_shaker.py
index 9a5e5a9..387abe0 100644
--- a/tests/test_shaker.py
+++ b/tests/test_shaker.py
@@ -13,7 +13,7 @@ def parse(tmp_path: Path, content: str) -> list:
     return getASTfromFile(str(f), process_includes=False) or []
 
 
-def make_pool(tmp_path: Path, content: str) -> dict:
+def make_pool(tmp_path: Path, content: str) -> dict[str, list]:
     """Parse content and return a list-valued pool dict (mirrors _add_to_pool logic)."""
     nodes = parse(tmp_path, content)
     pool: dict[str, list] = {}