Small yaml fix