diff --git a/pytest_localserver/smtp.py b/pytest_localserver/smtp.py
index 7779efb..00cd73e 100644
--- a/pytest_localserver/smtp.py
+++ b/pytest_localserver/smtp.py
@@ -156,7 +156,7 @@ def __repr__(self):  # pragma: no cover
         return "<smtp.Server %s:%s>" % self.addr
 
 
-if __name__ == "__main__":  # pragma: no cover
+def main():
     import time
 
     server = Server()
@@ -166,15 +166,14 @@ def __repr__(self):  # pragma: no cover
     print("Type <Ctrl-C> to stop")
 
     try:
-
-        try:
-            while True:
-                time.sleep(1)
-        finally:
-            print("\rstopping...")
-            server.stop()
-
+        while True:
+            time.sleep(1)
     except KeyboardInterrupt:
-        # support for Python 2.4 dictates that try ... finally is not used
-        # together with any except statements
         pass
+    finally:
+        print("\rstopping...")
+        server.stop()
+
+
+if __name__ == "__main__":  # pragma: no cover
+    main()