diff --git a/lib/components/NodeContainer.tsx b/lib/components/NodeContainer.tsx
index 7ae3fe2..4d7e98d 100644
--- a/lib/components/NodeContainer.tsx
+++ b/lib/components/NodeContainer.tsx
@@ -26,7 +26,7 @@ export function NodeContainer({
   children,
 }: NodeContainerProps) {
   const config = useGraphStore((store) => store.config)
-  const nodeConfig = config.getNodeConfig(node.type!)
+  const nodeConfig = config.getNodeConfig(node.type!)!
   const nodeGroupConfig = config.getNodeGroupConfig(nodeConfig.group)
   const [collapsed, toggleCollapsed] = useNodeCollapsed()