fixing the class