correct classes in the print output