fix forgot a class