Fix the dump_yaml task