Remove temporal line just before creating the new thought

1 files changed, 6 insertions, 2 deletions

diff --git a/view.py b/view.py
index 26725a1..3e5e02e 100644
--- a/view.py
+++ b/view.py
@@ -134,12 +134,16 @@ class NewThoughtTool(HandleTool):
     def ungrab_handle(self):
         logging.debug('NewThoughtTool.ungrab_handle')
         HandleTool.ungrab_handle(self)
-        self._new_connection = None
 
     def on_button_release(self, context, event):
-        HandleTool.on_button_release(self, context, event)
+        logging.debug('NewThoughtTool.ungrab_handle')
+
+        context.view.canvas.remove(self._new_connection)
+        self._new_connection = None
+
         context.view.model.create_new_thought(x=event.x, y=event.y,
                                               parent_id=self._parent_thought.id)
+        HandleTool.on_button_release(self, context, event)
 
     def move(self, view, item, handle, pos):
         #logging.debug('NewThoughtTool.move')