diff --git a/SketchWidget.py b/SketchWidget.py
index 741c26e..daaa497 100644
--- a/SketchWidget.py
+++ b/SketchWidget.py
@@ -9,8 +9,9 @@ from shapes import *
 from constraints import *
 
 class SketchWidget:
-    def __init__(self, sketch):
+    def __init__(self, sketch, window):
         self.sketch = sketch
+        self.window = window
         self.view_center = (0, 0)
         self.view_width = 4.0
         self.size = (1, 1)
diff --git a/Window.py b/Window.py
index e13782f..2ca18e7 100644
--- a/Window.py
+++ b/Window.py
@@ -52,7 +52,7 @@ class Window:
 
         s.solve()
 
-        sw = SketchWidget(s)
+        sw = SketchWidget(s, self.window)
         self.window.add(sw.widget)
         self.window.show_all()