diff options
Diffstat (limited to 'libmproxy/protocol2/layer.py')
| -rw-r--r-- | libmproxy/protocol2/layer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/libmproxy/protocol2/layer.py b/libmproxy/protocol2/layer.py index f2d6b3fb..8e985d4d 100644 --- a/libmproxy/protocol2/layer.py +++ b/libmproxy/protocol2/layer.py @@ -35,7 +35,7 @@ import threading from netlib import tcp from ..proxy import Log from ..proxy.connection import ServerConnection -from .messages import Connect, Reconnect, ChangeServer, Kill +from .messages import Connect, Reconnect, SetServer, Kill from ..exceptions import ProtocolException @@ -116,7 +116,7 @@ class ServerConnectionMixin(object): elif message == Connect: self._connect() return True - elif message == ChangeServer and message.depth == 1: + elif message == SetServer and message.depth == 1: if self.server_conn: self._disconnect() self.server_address = message.address |