diff --git a/cover_agent/CoverAgent.py b/cover_agent/CoverAgent.py
index ecefe65df..09c051968 100644
--- a/cover_agent/CoverAgent.py
+++ b/cover_agent/CoverAgent.py
@@ -205,7 +205,8 @@ def run(self):
             # Check if the desired coverage has been reached
             if self.test_validator.current_coverage < (self.test_validator.desired_coverage / 100):
                 # Run the coverage tool again if the desired coverage hasn't been reached
-                self.test_validator.run_coverage()
+                failed_test_runs, language, test_framework, coverage_report = self.test_validator.get_coverage()
+                self.test_gen.build_prompt(failed_test_runs, language, test_framework, coverage_report)
 
         # Log the final coverage
         if self.test_validator.current_coverage >= (self.test_validator.desired_coverage / 100):
diff --git a/cover_agent/version.txt b/cover_agent/version.txt
index a53741c09..967b33ffb 100644
--- a/cover_agent/version.txt
+++ b/cover_agent/version.txt
@@ -1 +1 @@
-0.2.6
\ No newline at end of file
+0.2.7
\ No newline at end of file