diff --git a/configs/parallelqa/tools.py b/configs/parallelqa/tools.py
index 08e8fad..36e64b2 100644
--- a/configs/parallelqa/tools.py
+++ b/configs/parallelqa/tools.py
@@ -81,7 +81,7 @@ async def run_llm_math_chain(question, context=None):
 def generate_tools(args, model_name):
     llm_math_chain = get_model(
         model_type=args.model_type,
-        api_key=args.api_key,
+        model_name=args.model_name,
         vllm_port=args.vllm_port,
         stream=False,
         temperature=0,