From f4f2f50364af5ec934eddbfc8bf994578d3bfd78 Mon Sep 17 00:00:00 2001
From: Roland Tannous <rolandtannous@gravityq.ai>
Date: Mon, 16 Feb 2026 17:34:52 +0000
Subject: [PATCH] fix: show Back to configuration breadcrumb when training
 completes

---
 studio/frontend/src/features/studio/studio-page.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/studio/frontend/src/features/studio/studio-page.tsx b/studio/frontend/src/features/studio/studio-page.tsx
index 8b92636fde..ebf4e3881b 100644
--- a/studio/frontend/src/features/studio/studio-page.tsx
+++ b/studio/frontend/src/features/studio/studio-page.tsx
@@ -36,7 +36,7 @@ export function StudioPage(): ReactElement {
   const dialogInitial = useDatasetPreviewDialogStore((s) => s.initialData);
   const closeDialog = useDatasetPreviewDialogStore((s) => s.close);
 
-  const canGoBack = runtimePhase === "stopped" || runtimePhase === "error";
+  const canGoBack = runtimePhase === "stopped" || runtimePhase === "error" || runtimePhase === "completed";
   const tourEnabled = hasHydratedRuntime && !isHydratingRuntime;
   const isConfigTour = !showTrainingView;
   const tourSteps = showTrainingView ? studioTrainingTourSteps : studioTourSteps;