Fix for heroku error