tiny refactor.