Fixed silly bug introduced by slight refactor.