Fix for Heroku deployment