Fix tests now we pass in the YAML file.

[#72786666]