fix web socket issues