Fix heroku trubl 2