diff --git a/packages/elements/src/node/Node.vue b/packages/elements/src/node/Node.vue
index d48d1f0..f4839bc 100644
--- a/packages/elements/src/node/Node.vue
+++ b/packages/elements/src/node/Node.vue
@@ -17,12 +17,14 @@ interface NodeProps {
 
 const props = defineProps<NodeProps>()
 const delegatedProps = reactiveOmit(props, 'class')
+
+const nodeClass = computed(() => cn('node-container relative size-full h-auto w-sm gap-0 rounded-md p-0', props.class))
 </script>
 
 <template>
   <Card
     v-bind="delegatedProps"
-    :class="cn('node-container relative size-full h-auto w-sm gap-0 rounded-md p-0', props.class)"
+    :class="nodeClass"
   >
     <Handle v-if="props.handles?.target" :position="Position.Left" type="target" />
     <Handle v-if="props.handles?.source" :position="Position.Right" type="source" />
diff --git a/packages/elements/src/node/NodeFooter.vue b/packages/elements/src/node/NodeFooter.vue
index c605678..575840d 100644
--- a/packages/elements/src/node/NodeFooter.vue
+++ b/packages/elements/src/node/NodeFooter.vue
@@ -6,10 +6,11 @@ import { cn } from '@repo/shadcn-vue/lib/utils'
 const props = defineProps<{
   class?: HTMLAttributes['class']
 }>()
+const className = computed(() => cn('rounded-b-md border-t bg-secondary p-3!', props.class))
 </script>
 
 <template>
-  <CardFooter :class="cn('rounded-b-md border-t bg-secondary p-3!', props.class)">
+  <CardFooter :class="className">
     <slot />
   </CardFooter>
 </template>
diff --git a/packages/elements/src/node/NodeHeader.vue b/packages/elements/src/node/NodeHeader.vue
index c984742..8df5e0b 100644
--- a/packages/elements/src/node/NodeHeader.vue
+++ b/packages/elements/src/node/NodeHeader.vue
@@ -6,10 +6,11 @@ import { cn } from '@repo/shadcn-vue/lib/utils'
 const props = defineProps<{
   class?: HTMLAttributes['class']
 }>()
+const className = computed(() => cn('gap-0.5 rounded-t-md border-b bg-secondary p-3!', props.class))
 </script>
 
 <template>
-  <CardHeader :class="cn('gap-0.5 rounded-t-md border-b bg-secondary p-3!', props.class)">
+  <CardHeader :class="className">
     <slot />
   </CardHeader>
 </template>