diff --git a/paperqa/docs.py b/paperqa/docs.py
index 367851b6..7adaeb5b 100644
--- a/paperqa/docs.py
+++ b/paperqa/docs.py
@@ -596,7 +596,7 @@ async def aquery(
         bib_str = "\n\n".join(
             [f"{i+1}. ({k}): {c}" for i, (k, c) in enumerate(bib.items())]
         )
-        formatted_answer = f"Question: {query}\n\n{answer_text}\n"
+        formatted_answer = f"Question: {answer.question}\n\n{answer_text}\n"
         if len(bib) > 0:
             formatted_answer += f"\nReferences\n\n{bib_str}\n"
         answer.answer = answer_text
@@ -611,7 +611,7 @@ async def aquery(
             )
             post = await chain.arun(**answer.dict(), callbacks=get_callbacks("post"))
             answer.answer = post
-            answer.formatted_answer = f"Question: {query}\n\n{post}\n"
+            answer.formatted_answer = f"Question: {answer.question}\n\n{post}\n"
             if len(bib) > 0:
                 answer.formatted_answer += f"\nReferences\n\n{bib_str}\n"
         if self.memory_model is not None:
diff --git a/paperqa/version.py b/paperqa/version.py
index 88c513ea..ff041687 100644
--- a/paperqa/version.py
+++ b/paperqa/version.py
@@ -1 +1 @@
-__version__ = "3.3.0"
+__version__ = "3.3.1"