diff --git a/apps/web/src/components/BranchToolbar.tsx b/apps/web/src/components/BranchToolbar.tsx
index e703427d4b6..2a1d0d1ec05 100644
--- a/apps/web/src/components/BranchToolbar.tsx
+++ b/apps/web/src/components/BranchToolbar.tsx
@@ -321,7 +321,7 @@ export const BranchToolbar = memo(function BranchToolbar({
onUsePreviousWorktree={onUsePreviousWorktree}
/>
) : (
-
+
{showEnvironmentIndicator && availableEnvironments && (
<>
+
{activeWorktreePath ? (
<>
@@ -79,7 +79,12 @@ export const BranchToolbarEnvModeSelector = memo(function BranchToolbarEnvModeSe
}}
items={envModeItems}
>
-
+
{effectiveEnvMode === "worktree" ? (
) : activeWorktreePath ? (
diff --git a/apps/web/src/components/BranchToolbarEnvironmentSelector.tsx b/apps/web/src/components/BranchToolbarEnvironmentSelector.tsx
index dbc742bea5a..e4ed54758ff 100644
--- a/apps/web/src/components/BranchToolbarEnvironmentSelector.tsx
+++ b/apps/web/src/components/BranchToolbarEnvironmentSelector.tsx
@@ -43,13 +43,13 @@ export const BranchToolbarEnvironmentSelector = memo(function BranchToolbarEnvir
if (envLocked || onEnvironmentChange === undefined) {
return (
-
+
{activeEnvironment?.isPrimary ? (
-
+
) : (
-
+
)}
- {activeEnvironment?.label ?? "Run on"}
+ {activeEnvironment?.label ?? "Run on"}
);
}
@@ -61,11 +61,16 @@ export const BranchToolbarEnvironmentSelector = memo(function BranchToolbarEnvir
onValueChange={(value) => onEnvironmentChange(value as EnvironmentId)}
items={environmentItems}
>
-
+
{activeEnvironment?.isPrimary ? (
-
+
) : (
-
+
)}