diff --git a/setup.py b/setup.py
index 6d9f62b..adb6d7c 100644
--- a/setup.py
+++ b/setup.py
@@ -79,7 +79,7 @@ def session_setup():
             print("toto")
     if uptime_days() > boot_limit:
         #message_box_ask_reboot(message_box_title, boot_limit_message_string % convert_time_to_date(uptime_seconds(), translate=True))
-        if ask_message(message_box_title, boot_limit_message_string % convert_time_to_date(uptime_seconds(), translate=True), flags=4, raise_error=False,) == "6" :
+        if ask_message(message_box_title, boot_limit_message_string % convert_time_to_date(uptime_seconds(), translate=True), flags=4, raise_error=False,) == 6 :
             reboot_machine()
         return "WARNING"
     if uptime_days() < boot_limit: