Fix test and update mockito