fixed mockito after refactor