Fixing undefined YAML error
Merge pull request #32 from hron84/fix-tests

Fixing undefined YAML error