diff --git a/packages/ai-workspace-common/src/components/canvas/node-preview/node-preview-header.tsx b/packages/ai-workspace-common/src/components/canvas/node-preview/node-preview-header.tsx
index bdf48df29..5908d046a 100644
--- a/packages/ai-workspace-common/src/components/canvas/node-preview/node-preview-header.tsx
+++ b/packages/ai-workspace-common/src/components/canvas/node-preview/node-preview-header.tsx
@@ -7,6 +7,7 @@ import {
   Sparkles,
   Wrench,
   Maximize2,
+  Minimize2,
   MoreHorizontal,
   X,
   Cpu,
@@ -258,7 +259,7 @@ export const NodePreviewHeader: FC<NodePreviewHeaderProps> = ({
             className={`p-1.5 hover:bg-gray-100 ${isMaximized ? 'text-primary-600' : 'text-gray-500'}`}
             onClick={() => onMaximize()}
           >
-            <Maximize2 className="w-4 h-4" />
+            {isMaximized ? <Minimize2 className="w-4 h-4" /> : <Maximize2 className="w-4 h-4" />}
           </Button>
         )}
         <Button