Add missing mock