Fix unfullscreen unfocus bug

1 files changed, 5 insertions, 8 deletions

diff --git a/hello-integration.py b/hello-integration.py
index ba77581..05b8617 100755
--- a/hello-integration.py
+++ b/hello-integration.py
@@ -132,10 +132,6 @@ class GraphPlotter(gtk.Window):
         self._vbox.show()
 
         self._is_fullscreen = False
-        self._unfullscreen_button = UnfullscreenButton()
-        self._unfullscreen_button.set_transient_for(self)
-        self._unfullscreen_button.connect_button_press(
-            lambda w, e: self.unfullscreen())
         self.connect('motion-notify-event', self.motion_cb)
         self.connect('expose-event', self.expose_event)
 
@@ -161,17 +157,18 @@ class GraphPlotter(gtk.Window):
 
     def fullscreen(self):
         self.options.hide()
-
         self._is_fullscreen = True
-
+        self._unfullscreen_button = UnfullscreenButton()
+        self._unfullscreen_button.set_transient_for(self)
+        self._unfullscreen_button.connect_button_press(
+            lambda w, e: self.unfullscreen())
         self._unfullscreen_button.show()
         gtk.Window.fullscreen(self)
 
     def unfullscreen(self):
         self.options.show()
         self._is_fullscreen = False
-
-        self._unfullscreen_button.hide()
+        self._unfullscreen_button.destroy()
         gtk.Window.unfullscreen(self)
 
     def export(self, widget, data):