fixed minor error in code