use correct classes for model