diff --git a/net/ws.go b/net/ws.go
index acf8112..4eef5ed 100644
--- a/net/ws.go
+++ b/net/ws.go
@@ -154,6 +154,10 @@ func (p *WebSocketPeer) GetIP() net.Addr {
 	return ipAddr
 }
 
+func (p *WebSocketPeer) SetIP(ip net.Addr) {
+	p.ip = ip
+}
+
 func (b *WebSocketPeer) GetIPString() string {
 	return b.GetIP().String()
 }