Fixing the yaml