add missing classes for previous commit