fix error where classes were getting the wrong name