diff --git a/src/App.tsx b/src/App.tsx
index 5c877ca..913c777 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -173,7 +173,7 @@ function AppContent() {
                 setSelectedNode(null);
               }
             }}
-            onAddNodeRequest={(callback: any) => setAddNodeCallback(() => callback)}
+            onAddNodeRequest={(callback: (nodeTypeId: string, position?: { x: number; y: number }) => void) => setAddNodeCallback(() => callback)}
           />
         </div>
 
diff --git a/src/components/Editor/GraphEditor.tsx b/src/components/Editor/GraphEditor.tsx
index 18ca03e..439b00e 100644
--- a/src/components/Editor/GraphEditor.tsx
+++ b/src/components/Editor/GraphEditor.tsx
@@ -503,7 +503,7 @@ const GraphEditor = ({ onNodeSelect, onEdgeSelect, onAddNodeRequest }: GraphEdit
   // Call the onAddNodeRequest callback if provided
   useEffect(() => {
     if (onAddNodeRequest) {
-      onAddNodeRequest(handleAddNode as any);
+      onAddNodeRequest(handleAddNode);
     }
   }, [onAddNodeRequest, handleAddNode]);