diff --git a/navi/core/agent.py b/navi/core/agent.py
index 72363eb..7b6b716 100644
--- a/navi/core/agent.py
+++ b/navi/core/agent.py
@@ -864,7 +864,7 @@
         phase2_ctx: list[Message] = [phase2_system]
         if mem:
             phase2_ctx.append(mem)
-        user_msgs = [m for m in session.context if m.role == "user"]
+        user_msgs = [m for m in context if m.role == "user"]
         if user_msgs:
             phase2_ctx.append(user_msgs[-1])