1 files changed, 5 insertions, 14 deletions
diff --git a/src/gui/MachinaCanvas.hpp b/src/gui/MachinaCanvas.hpp
index c396931..52e2d74 100644
--- a/src/gui/MachinaCanvas.hpp
+++ b/src/gui/MachinaCanvas.hpp
@@ -40,9 +40,6 @@ class MachinaCanvas
 public:
 	MachinaCanvas(MachinaGUI* app, int width, int height);
 
-	//void build(SPtr<const machina::Machine> machine, bool show_labels);
-	//void update_edges();
-
 	void on_new_object(SPtr<machina::client::ClientObject> object);
 	void on_erase_object(SPtr<machina::client::ClientObject> object);
 
@@ -51,21 +48,15 @@ public:
 protected:
 	bool on_event(GdkEvent* event);
 
-	bool node_clicked(WPtr<NodeView> item, GdkEventButton* ev);
+	bool node_clicked(NodeView* node, GdkEventButton* ev);
 
 private:
-	//SPtr<NodeView> create_node_view(SPtr<machina::Node> node);
-
 	void action_create_node(double x, double y);
+	void action_connect(NodeView* tail, NodeView* head);
+	void action_disconnect(NodeView* tail, NodeView* head);
 
-	void action_connect(SPtr<NodeView> port1,
-	                    SPtr<NodeView> port2);
-
-	void action_disconnect(SPtr<NodeView> port1,
-	                       SPtr<NodeView> port2);
-
-	MachinaGUI*    _app;
-	WPtr<NodeView> _last_clicked;
+	MachinaGUI* _app;
+	NodeView*   _last_clicked;
 };
 
 }  // namespace machina