Fix test mistakes