1 files changed, 8 insertions, 4 deletions

diff --git a/renderer/src/App.tsx b/renderer/src/App.tsx
index ecbb5b2..74b1f91 100644
--- a/renderer/src/App.tsx
+++ b/renderer/src/App.tsx
@@ -263,13 +263,17 @@ export function App() {
         selectedSession.phase === "research" ? "research.md" : "plan.md";
       await api.writeSessionArtifact(selectedProject.id, selectedSession.id, filename, documentContent);
 
-      const nextPhase = await api.advancePhase(selectedSession.id);
-      if (nextPhase) {
-        setSelectedSession({ ...selectedSession, phase: nextPhase });
+      const advanced = await api.advancePhase(selectedSession.id);
+      if (advanced) {
+        setSelectedSession({
+          ...selectedSession,
+          phase: advanced.phase,
+          git_branch: advanced.git_branch,
+        });
         // Trigger initial message for next phase
         setIsLoading(true);
         const initialMsg =
-          nextPhase === "plan"
+          advanced.phase === "plan"
             ? "Create a detailed implementation plan based on the research."
             : "Begin implementing the plan.";
         await api.sendMessage(selectedSession.id, initialMsg);