Fixing web sockets issue