diff --git a/flask_sockets.py b/flask_sockets.py index 2670091..c0b4b64 100644 --- a/flask_sockets.py +++ b/flask_sockets.py @@ -38,7 +38,7 @@ class SocketMiddleware(object): environment = environ['wsgi.websocket'] handler(environment, **values) return [] - except NotFound: + except (NotFound, KeyError): return self.app(environ, start_response)