Fix misplaced class