Bug fix for heroku deployment.