Fixed bugs when run on Heroku.