Add missing mock function.