Updated YAML to fix bug in #501