Add mock to test deps