diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py
index 3ec460a..eae74a3 100644
--- a/src/options/generate/generator.py
+++ b/src/options/generate/generator.py
@@ -519,8 +519,9 @@ Test scenario:
                 setattr(self.microservice_specification, refinement_type, final)
                 break
             elif agent_question:
-                messages.append(AIMessage(content=agent_question))
-                user_input = self.get_user_input(pm, agent_question)
+                question_parsed = json.loads(agent_question)['question']
+                messages.append(AIMessage(content=question_parsed))
+                user_input = self.get_user_input(pm, question_parsed)
             else:
                 if num_parsing_tries > 2:
                     raise self.TaskRefinementException()
diff --git a/test/test_generator.py b/test/test_generator.py
index 54c7573..50bc50c 100644
--- a/test/test_generator.py
+++ b/test/test_generator.py
@@ -85,7 +85,7 @@ def test_generation_level_3(tmpdir):
     """
     os.environ['VERBOSE'] = 'true'
     generator = Generator(
-        f'''Given an audio file of speech like https://www.signalogic.com/melp/EngSamples/Orig/ENG_M.wav, 
+        f'''Given an audio file (1min) of speech like https://www.signalogic.com/melp/EngSamples/Orig/ENG_M.wav, 
 get convert it to text using the following api:
 import requests
 url = "https://transcribe.whisperapi.com"