Fix test Mock error