diff --git a/src/ws/init.go b/src/ws/init.go
index 3dbbfd5..e6993ff 100644
--- a/src/ws/init.go
+++ b/src/ws/init.go
@@ -44,10 +44,8 @@ func welcomeAgents(ws *websocket.Conn, welcome wsWelcome, t string) {
 
 func gcConnOnError(ws *websocket.Conn) (err error) {
 	listeners.Range(func(index, value interface{}) bool {
-		if value.(*connectionInfo).ConnectionPS == ws {
+		if value.(*connectionInfo).ConnectionPS == ws || value.(*connectionInfo).ConnectionRR == ws {
 			value.(*connectionInfo).ConnectionPS.Close()
-			listeners.Delete(index)
-		} else if value.(*connectionInfo).ConnectionRR == ws {
 			value.(*connectionInfo).ConnectionRR.Close()
 			listeners.Delete(index)
 		}