Added support for partial mocks