Fix bug in controller. It's Sunday afternoon and I neglect to do proper TDD.