fixes names of classes, ugh