diff --git a/dot-to-dot b/dot-to-dot
index 10c8fdc..13c0274 100755
Binary files a/dot-to-dot and b/dot-to-dot differ
diff --git a/imperative-shell/event-loop.sml b/imperative-shell/event-loop.sml
index 647ffa7..43c6913 100644
--- a/imperative-shell/event-loop.sml
+++ b/imperative-shell/event-loop.sml
@@ -17,7 +17,7 @@ struct
       loop (inputMailbox, drawMailbox, 0, 0, AppType.initial)
   end
 
-  fun draw (window, graphDrawObject, buttonDrawObject, buttonDrawLength) =
+  fun draw (drawMailbox, window, graphDrawObject, buttonDrawObject, buttonDrawLength) =
     if not (Glfw.windowShouldClose window) then
       let
         val _ = Gles3.clearColor (1.0, 1.0, 1.0, 1.0)
@@ -29,7 +29,7 @@ struct
         val _ = Glfw.pollEvents ()
         val _ = Glfw.swapBuffers window
       in
-        draw (window, graphDrawObject, buttonDrawObject, buttonDrawLength)
+        draw (drawMailbox, window, graphDrawObject, buttonDrawObject, buttonDrawLength)
       end
     else
       Glfw.terminate ()
diff --git a/imperative-shell/shell.sml b/imperative-shell/shell.sml
index 0fe5d03..a5e8bb1 100644
--- a/imperative-shell/shell.sml
+++ b/imperative-shell/shell.sml
@@ -21,7 +21,7 @@ struct
 
       val _ = CML.spawn (fn () => InputCallbacks.registerCallbacks (window, inputMailbox))
       val _ = CML.spawn (fn () => EventLoop.update (inputMailbox, drawMailbox))
-      val _ = CML.spawn (fn () => EventLoop.draw (window, graphDrawObject, buttonDrawObject, 0))
+      val _ = CML.spawn (fn () => EventLoop.draw (drawMailbox, window, graphDrawObject, buttonDrawObject, 0))
     in
       ()
     end