Fix YAML test failure

Added YAML require to prevent test suite from blowing up.