diff --git a/tests/test_factor_rules.py b/tests/test_factor_rules.py
index d40dfb5d..78d45508 100644
--- a/tests/test_factor_rules.py
+++ b/tests/test_factor_rules.py
@@ -78,7 +78,7 @@ class RecordingLLM:
 
 
 def make_factory(recorded_llms, node_name):
-    def factory(llm, *_args):
+    def factory(llm, *_args, **_kwargs):
         recorded_llms[node_name] = llm
         return node_name