Fix test .yaml file with latest imports