diff --git a/server.py b/server.py
index 430d1dc7..a6f38e65 100644
--- a/server.py
+++ b/server.py
@@ -6,14 +6,15 @@
 
 
 class Handler(http.server.SimpleHTTPRequestHandler):
-    def do_GET(self):
+    def __enter__(self):
         self.send_response(HTTPStatus.OK)
         self.end_headers()
         msg = 'Hello! you requested %s' % (self.path)
         self.wfile.write(msg.encode())
 
 
-port = int(os.getenv('PORT', 80))
-print('Listening on port %s' % (port))
-httpd = socketserver.TCPServer(('', port), Handler)
-httpd.serve_forever()
+if __name__ == "__main__":
+    port = int(os.getenv('PORT', 80))
+    print('Listening on port %s' % (port))
+    httpd = socketserver.TCPServer(('', port), Handler)
+    httpd.serve_forever()