Minor modification for heroku deployment