fix typo to assign classes