Fixing heroku support