Fixed heroku issue