Fix method names in mock