Minor refactoring change to fix up a method.