Fix mockito deprecation warnings