little tests refactor