Unit tests refactor.