Fix Heroku error