+ Added missing mock method