Fix minor error in formulation