Turn of design for extension because Mockito cannot support it