Fix yaml bug