Update sock to receive messages from server properly