diff --git a/wsproxy/websocket_proxy.go b/wsproxy/websocket_proxy.go
index 7092162..009b724 100644
--- a/wsproxy/websocket_proxy.go
+++ b/wsproxy/websocket_proxy.go
@@ -294,6 +294,9 @@ func (p *Proxy) proxy(w http.ResponseWriter, r *http.Request) {
 	}
 
 	for scanner.Scan() {
+		if p.pingWait > 0 {
+			conn.SetWriteDeadline(time.Now().Add(p.pingWait))
+		}
 		if len(scanner.Bytes()) == 0 {
 			p.logger.Warnln("[write] empty scan", scanner.Err())
 			continue