Move to mocking framework