diff --git a/src/components/graph/BaseNode.tsx b/src/components/graph/BaseNode.tsx
index 58d01b5..4ce646f 100644
--- a/src/components/graph/BaseNode.tsx
+++ b/src/components/graph/BaseNode.tsx
@@ -124,7 +124,7 @@ export function BaseNodeHeaderRow({
   return (
     <header
       className={cn(
-        "shrink-0 mx-0 my-0 -mb-1 flex flex-row items-center justify-between gap-2 border-b px-3 py-2",
+        "shrink-0 mx-0 my-0 flex flex-row items-center justify-between gap-2 px-3 py-2",
         className,
       )}
     >
@@ -162,7 +162,7 @@ export function BaseNodeContent({
   return (
     <div
       data-slot="base-node-content"
-      className={cn("min-h-0 flex-1 flex flex-col overflow-auto pt-1", className)}
+      className={cn("min-h-0 flex-1 flex flex-col overflow-auto", className)}
       {...props}
     />
   );
diff --git a/src/components/graph/NodeHandles.tsx b/src/components/graph/NodeHandles.tsx
index 2e11663..293dcf9 100644
--- a/src/components/graph/NodeHandles.tsx
+++ b/src/components/graph/NodeHandles.tsx
@@ -1,6 +1,6 @@
 import React from 'react'
 import { Handle, Position } from '@xyflow/react'
-import { Circle, CircleDashed } from 'lucide-react'
+import { ArrowDownLeft, ArrowUpRight } from 'lucide-react'
 
 type NodeHandleProps = {
     id: string
@@ -26,7 +26,7 @@ export function InputHandle({ id }: NodeHandleProps) {
                 justifyContent: 'center',
             }}
         >
-            <CircleDashed className="w-4 h-4 text-muted-foreground pointer-events-none" />
+            <ArrowDownLeft className="w-4 h-4 text-muted-foreground pointer-events-none" />
         </Handle>
     )
 }
@@ -51,7 +51,7 @@ export function OutputHandle({ id }: NodeHandleProps) {
                 justifyContent: 'center',
             }}
         >
-            <Circle className="w-4 h-4 text-foreground pointer-events-none" />
+            <ArrowUpRight className="w-4 h-4 text-foreground pointer-events-none" />
         </Handle>
     )
 }