I think you can drop the class_name here