Fix YAML test failure
Merge pull request #71 from agraves/fix_yaml_failure

Fix YAML test failure