Fix an out-of-date mock