1 files changed, 3 insertions, 3 deletions
diff --git a/src/main.c b/src/main.c
index 7a8c19a..a41973b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -390,21 +390,21 @@ main (int argc, char *argv[])
     {
       send_kill ();
       XSync (dpy, False);
-      clean_up ();
+      XCloseDisplay (dpy);
       return EXIT_SUCCESS;
     }
   if (do_restart)
     {
       send_restart ();
       XSync (dpy, False);
-      clean_up ();
+      XCloseDisplay (dpy);
       return EXIT_SUCCESS;
     }
   if (do_command)
     {
       send_command (command);
       free (command);
-      clean_up();
+      XCloseDisplay (dpy);
       return EXIT_SUCCESS;
     }