Fix bug class_name != class_dt