fixed one error  in class