Add mock to test.