small code change in class definition