little tiny refactor