Fix heroku startup error introduced by refactor in previous commit