Fix heroku bug