Fix for Heroku deploy