Fixes YAML Mapping issue from #11. Thanks @LeonB!