fix yaml mode
Merge pull request #521 from daeks/yaml

Fix: yaml mode