Small tests refactor.