diff --git a/egui_node_graph/src/editor_ui.rs b/egui_node_graph/src/editor_ui.rs
index b1fc5fd..b955111 100644
--- a/egui_node_graph/src/editor_ui.rs
+++ b/egui_node_graph/src/editor_ui.rs
@@ -292,7 +292,7 @@ where
                     self.node_order.retain(|id| *id != *node_id);
                 }
                 NodeResponse::DisconnectEvent { input, output } => {
-                    let other_node = self.graph.get_input(*input).node();
+                    let other_node = self.graph.get_output(*output).node;
                     self.graph.remove_connection(*input);
                     self.connection_in_progress =
                         Some((other_node, AnyParameterId::Output(*output)));