Fix failing yaml test

On top of #21