Update the example yaml to match current functionality