Fix for Heroku push