Mockito fix