Fixing yaml issue