diff --git a/app/w/[id]/components/workflow-block/components/sub-block/components/code.tsx b/app/w/[id]/components/workflow-block/components/sub-block/components/code.tsx
index 842dea2c31..3971150676 100644
--- a/app/w/[id]/components/workflow-block/components/sub-block/components/code.tsx
+++ b/app/w/[id]/components/workflow-block/components/sub-block/components/code.tsx
@@ -196,11 +196,11 @@ export function Code({ blockId, subBlockId, isConnecting, minimizable }: CodePro
       onDragOver={(e) => e.preventDefault()}
       onDrop={handleDrop}
     >
-      {minimizable && code.length > 0 && (
+      {minimizable && code.length > 0 && lineCount > 3 && (
         <Button
           variant="ghost"
           size="icon"
-          className="absolute right-2 top-2 z-10 h-6 w-6 hover:bg-accent hover:text-accent-foreground"
+          className="absolute right-2 top-2 z-10 h-6 w-6 text-muted-foreground hover:bg-muted hover:text-foreground"
           onClick={() => setIsMinimized(!isMinimized)}
         >
           {isMinimized ? <ChevronDown className="h-4 w-4" /> : <ChevronUp className="h-4 w-4" />}
diff --git a/blocks/blocks/condition.ts b/blocks/blocks/condition.ts
index ef8e4846d6..e09852afd4 100644
--- a/blocks/blocks/condition.ts
+++ b/blocks/blocks/condition.ts
@@ -32,21 +32,6 @@ export const ConditionBlock: BlockConfig<CodeExecutionOutput> = {
         title: 'if',
         type: 'code',
         layout: 'full',
-        outputHandle: true,
-      },
-      {
-        id: 'else-if',
-        title: 'else if',
-        type: 'code',
-        layout: 'full',
-        outputHandle: true,
-      },
-      {
-        id: 'else',
-        title: 'else',
-        type: 'code',
-        layout: 'full',
-        outputHandle: true,
       },
     ],
   },