added a fix for heroku