diff --git a/src/components/Canvas/FlowCanvas/components/node.vue b/src/components/Canvas/FlowCanvas/components/node.vue
index ceffc32..d51161b 100644
--- a/src/components/Canvas/FlowCanvas/components/node.vue
+++ b/src/components/Canvas/FlowCanvas/components/node.vue
@@ -1,5 +1,5 @@
 <template>
-	<div class="node start" :class="{ center: posCenter }">
+	<div class="node" :class="{ center: posCenter }">
 		<Handle
 			v-for="handle in handles[type]"
 			:key="handle.id"
diff --git a/src/components/Canvas/FlowCanvas/manager/StateManager.ts b/src/components/Canvas/FlowCanvas/manager/StateManager.ts
index 7795e4c..de2163a 100644
--- a/src/components/Canvas/FlowCanvas/manager/StateManager.ts
+++ b/src/components/Canvas/FlowCanvas/manager/StateManager.ts
@@ -33,13 +33,10 @@ export class StateManager {
 					...node,
 				}
 				if (index === 0) {
-					obj.class = 'custom-node start';
 					obj.type = NODE_TYPE.INPUT;
 				} else if (index === this.nodes.value.length - 1) {
-					obj.class = 'custom-node end';
 					obj.type = NODE_TYPE.OUTPUT;
 				} else {
-					obj.class = 'custom-node';
 					obj.type = NODE_TYPE.SECONDARY;
 				}
 				return obj