diff --git a/ui/src/components/AgentControl.tsx b/ui/src/components/AgentControl.tsx
index 5484059..1ae77b3 100644
--- a/ui/src/components/AgentControl.tsx
+++ b/ui/src/components/AgentControl.tsx
@@ -1,4 +1,4 @@
-import { Play, Square, Loader2, Flame } from 'lucide-react'
+import { Play, Square, Loader2 } from 'lucide-react'
import {
useStartAgent,
useStopAgent,
@@ -40,8 +40,6 @@ export function AgentControl({ projectName, status }: AgentControlProps) {
>
{isLoading ? (
- ) : yoloMode ? (
-
) : (
)}
@@ -50,9 +48,11 @@ export function AgentControl({ projectName, status }: AgentControlProps) {