Fix YAML test failure