Fix bug in heroku