Added missing mock methods