diff --git a/site/src/pages/AgentsPage/components/WorkspacePill.tsx b/site/src/pages/AgentsPage/components/WorkspacePill.tsx
index 2ad1798033..51d291dcc4 100644
--- a/site/src/pages/AgentsPage/components/WorkspacePill.tsx
+++ b/site/src/pages/AgentsPage/components/WorkspacePill.tsx
@@ -1,7 +1,6 @@
import {
ChevronDownIcon,
CopyIcon,
- ExternalLinkIcon,
LayoutGridIcon,
MonitorIcon,
SquareTerminalIcon,
@@ -26,6 +25,8 @@ import {
DropdownMenuTrigger,
} from "#/components/DropdownMenu/DropdownMenu";
import { ExternalImage } from "#/components/ExternalImage/ExternalImage";
+import { VSCodeIcon } from "#/components/Icons/VSCodeIcon";
+import { VSCodeInsidersIcon } from "#/components/Icons/VSCodeInsidersIcon";
import {
Tooltip,
TooltipContent,
@@ -220,7 +221,11 @@ const VSCodeMenuItem: FC<{
onSelect={handleClick}
disabled={isGeneratingKey || !isRunning}
>
-
+ {variant === "vscode" ? (
+
+ ) : (
+
+ )}
{label}
);