Fix heroku error