Fix test where the absence of mock breaks things