Forgot the change to the yaml structure