Fixed unit test failing after mockito upgrade