Update socket to work with heroku