diff --git a/src/ws/init.go b/src/ws/init.go
index 2202a40..25fb6ae 100644
--- a/src/ws/init.go
+++ b/src/ws/init.go
@@ -21,13 +21,14 @@ func WelcomeAgents(ws *websocket.Conn, welcome WSWelcome, t string) {
 	connectinfo, ok := LISTENERS[welcome.Hostname]
 
 	if !ok {
-		if t == "ps" {
+		switch t {
+		case "ps":
 			connectinfo := ConnectionInfo{
 				ConnectionPS: ws,
 				InitDate:     time.Now(),
 			}
 			LISTENERS[welcome.Hostname] = &connectinfo
-		} else if t == "rr" {
+		case "rr":
 			connectinfo := ConnectionInfo{
 				ConnectionRR: ws,
 				InitDate:     time.Now(),
@@ -35,9 +36,10 @@ func WelcomeAgents(ws *websocket.Conn, welcome WSWelcome, t string) {
 			LISTENERS[welcome.Hostname] = &connectinfo
 		}
 	} else {
-		if t == "ps" {
+		switch t {
+		case "ps":
 			connectinfo.ConnectionPS = ws
-		} else if t == "rr" {
+		case "rr":
 			connectinfo.ConnectionRR = ws
 		}
 	}