Fix YAML definition