Modified socket tests to connect to Heroku