Changed some mockito code with jmock