Fixing heroku 2