Fix tests
Update yaml sample