diff --git a/ipe/websocket.go b/ipe/websocket.go
index 4d8d429..57cc65b 100644
--- a/ipe/websocket.go
+++ b/ipe/websocket.go
@@ -79,10 +79,11 @@ func onMessage(conn *websocket.Conn, w http.ResponseWriter, r *http.Request, ses
 
 		if err != nil {
 			log.Errorf("%+v", err)
-			switch err {
-			case io.EOF:
+			if err == io.EOF {
 				onClose(sessionID, app)
-			default:
+			} else if _, ok := err.(*websocket.CloseError); ok {
+				onClose(sessionID, app)
+			} else {
 				emitWSError(newGenericReconnectImmediatelyError(), conn)
 			}
 			break