Fix test after last push