Small fix.  Forgot to refactor a method