Fix test again