From b159ecbeb37b643abf81799f9a40706589faaf5a Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Wed, 3 Feb 2016 22:03:41 -0500
Subject: [PATCH] Fallback to Flask routing when not a ws request #11

---
 flask_sockets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)