Fixed socket bug (maybe?)