minor error in code