Fix yz tests