1 files changed, 6 insertions, 5 deletions
diff --git a/libmproxy/protocol2/reverse_proxy.py b/libmproxy/protocol2/reverse_proxy.py
index 767107ad..9d5a4beb 100644
--- a/libmproxy/protocol2/reverse_proxy.py
+++ b/libmproxy/protocol2/reverse_proxy.py
@@ -16,8 +16,9 @@ class ReverseProxy(Layer, ServerConnectionMixin):
             layer = TlsLayer(self, self._client_tls, self._server_tls)
         else:
             layer = self.ctx.next_layer(self)
-        for message in layer():
-            if not self._handle_server_message(message):
-                yield message
-        if self.server_conn:
-            self._disconnect()
-\ No newline at end of file
+
+        try:
+            layer()
+        finally:
+            if self.server_conn:
+                self._disconnect()
+\ No newline at end of file