Adding custom mock to fix build.