Fix error in heroku