Fix IMAP test