diff --git a/app/imap_connect.py b/app/imap_connect.py
index 2335f63..c9688aa 100644
--- a/app/imap_connect.py
+++ b/app/imap_connect.py
@@ -78,9 +78,7 @@ def run():
                 try:
                     logging.debug("Idle check for folder: {}".format(folder))
                     messages = _server.idle_check(timeout=60)  # Timeout after 60 seconds
-                    # if not messages:
-                    #     logging.info("No new messages in folder: {}".format(folder))
-                    #     continue
+                    _server.idle_done()
                     for item in messages:
                         if item[1] in (b'EXISTS'):
                             logging.info("New messages in folder: {}".format(folder))
@@ -89,7 +87,6 @@ def run():
                                     eventHandler('NewMailEvent', uid, message_data, _server)                    
                 except Exception as e:
                     logging.error("Error during idle check: {}".format(e))
-                _server.idle_done()
 
     username = os.environ.get('username')
     password = os.environ.get('password')