Fix yaml file to comply with previous refactoring