diff --git a/agents/dummy_prompt_agent.py b/agents/dummy_prompt_agent.py
index 9dc7596f267c409ff19dde8e76b6d9ca5b710c85..b7e5ba0075d703885e2f075d90b96d27442e2869 100644
--- a/agents/dummy_prompt_agent.py
+++ b/agents/dummy_prompt_agent.py
@@ -10,7 +10,7 @@ class DummyPromptAgent(object):
         self.max_generated_token_per_call = 20      # Can be set by user as needed, can be different for each utterance and dialogue
         self.api_usage_count = 0 
 
-    def generate_responses(self, test_data: List[Dict], api_responses: List[str]) -> List[str]:
+    def generate_responses(self, test_data: List[Dict], api_responses: List[str], final=False) -> List[str]:
         """
         You will be provided with a batch of upto 50 independent conversations
         
@@ -47,7 +47,7 @@ class DummyPromptAgent(object):
         """
         # print(f"{len(test_data)=}, {test_data[0].keys()=}, {len(test_data[-1]['dialogue'])}")
 
-        if self.api_usage_count < self.api_limit:
+        if self.api_usage_count < self.api_limit and not final:
             self.api_usage_count += 1
             response = {
                 "use_api": True,