Ensuring when can handle mocks too.