fixed a distraction error related to the classes used