change class to model.self