Fix yaml file for new format