little cosmetic refactor