Fix YAML loading
Merge pull request #3 from uber/yaml_fix

Fix YAML loading