Fix yaml to try and get it working