Fixed Socket.io integration