Fix  yaml bug