[fix] Added entity class.