1 files changed, 11 insertions, 2 deletions
diff --git a/src/wayclock.cc b/src/wayclock.cc
index 32d7023..369657e 100644
--- a/src/wayclock.cc
+++ b/src/wayclock.cc
@@ -3,9 +3,19 @@
 
 #define WAYC_APP_NAME "Wayclock"
 
+void handle(struct eventloop_s *loop, struct event_s *event) {
+  switch (event->kind) {
+  case EVENT_KIND_CLOSE:
+    wayc_eventloop_unregister(loop, event->window->id);
+    break;
+  default:
+    break;
+  }
+}
+
 int main() {
   struct eventloop_s loop;
-  if (!wayc_eventloop_init(&loop))
+  if (!wayc_eventloop_init(&loop, handle))
     wayc_panic("Failed to initialize event loop");
 
   struct window_s window;
@@ -19,7 +29,6 @@ int main() {
 
   while (wayc_eventloop_running(&loop)) {
     wayc_eventloop_update(&loop);
-    wayc_eventloop_unregister(&loop, winid);
   }
 
   wayc_window_deinit(&window);