Fixing mockito reference