diff --git a/Lib/test/test_imaplib.py b/Lib/test/test_imaplib.py
index 4b12d23..7db3f7d 100644
--- a/Lib/test/test_imaplib.py
+++ b/Lib/test/test_imaplib.py
@@ -444,7 +444,7 @@
                               keyfile=CERTFILE, ssl_context=self.create_ssl_context())
 
 
-def test_main():
+def load_tests(*args):
     tests = [TestImaplib]
 
     if support.is_resource_enabled('network'):
@@ -459,9 +459,9 @@
             RemoteIMAPTest, RemoteIMAP_SSLTest, RemoteIMAP_STARTTLSTest,
         ])
 
-    support.run_unittest(*tests)
+    return unittest.TestSuite([unittest.makeSuite(test) for test in tests])
 
 
 if __name__ == "__main__":
     support.use_resources = ['network']
-    test_main()
+    unittest.main()
diff --git a/Misc/NEWS b/Misc/NEWS
index 91c4153..83c38ec 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -644,6 +644,9 @@
 
 - Issue #15539: Added regression tests for Tools/scripts/pindent.py.
 
+- Issue #17333: test_imaplib now works with unittest test discovery.
+  Patch by Zachary Ware.
+
 - Issue #17082: test_dbm* now work with unittest test discovery.
   Patch by Zachary Ware.