Making this mock work