Fixes reference to Mockito