Fix for heroku deployment