Fixed UI tests after MVP refactor.