Fixed minor error in code