Remove mockito dependency