Fix heroku issue