Remove explicit class_name reference on models