diff options
| -rw-r--r-- | mitmproxy/eventsequence.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mitmproxy/eventsequence.py b/mitmproxy/eventsequence.py index 09322a8f..68d97a23 100644 --- a/mitmproxy/eventsequence.py +++ b/mitmproxy/eventsequence.py @@ -84,7 +84,7 @@ _iterate_map = { http.HTTPFlow: _iterate_http, websocket.WebSocketFlow: _iterate_websocket, tcp.TCPFlow: _iterate_tcp -} # type: typing.Dict[flow.Flow, typing.Callable[[flow.Flow], TEventGenerator]] +} # type: typing.Dict[typing.Type[flow.Flow], typing.Callable[[flow.Flow], TEventGenerator]] def iterate(f: flow.Flow) -> TEventGenerator: |