diff --git a/session/forwarder/forwarder.go b/session/forwarder/forwarder.go index 462df4b..c0330f8 100644 --- a/session/forwarder/forwarder.go +++ b/session/forwarder/forwarder.go @@ -171,7 +171,7 @@ func (f *Forwarder) WriteBadGatewayResponse(dst io.Writer) { } func (f *Forwarder) Close() error { - if f.GetTunnelType() != types.HTTP { + if f.GetListener() != nil { return f.Listener.Close() } return nil