Replace custom Stubs in favor of Mockito