tiny refactor
tiny refactor