Fixed issue with socket.io