fix incorrect classes