diff --git a/evoprompt/optimization.py b/evoprompt/optimization.py
index e27ce3d16bb98985a10a91da7d3837b5d715da77..6aa71e736ea3a9a080b77b43618be1ed248769d4 100644
--- a/evoprompt/optimization.py
+++ b/evoprompt/optimization.py
@@ -129,13 +129,15 @@ class PromptOptimization:
             initial_population += paraphrases
             logger.info(
                 "Paraphrased prompt '%s': %s.",
-                self.task.base_prompts[0].replace("\r", "\\r").replace("\n", "\\n"),
+                top_prompts[promptindex_to_paraphrase]
+                .replace("\r", "\\r")
+                .replace("\n", "\\n"),
                 paraphrases,
             )
             promptindex_to_paraphrase += 1
             promptindex_to_paraphrase %= len(top_prompts)
 
-        return self.task.base_prompts + paraphrases
+        return initial_population
 
     def add_prompt(
         self,