DocumentItem: fix inapropriate heritage

1 files changed, 5 insertions, 5 deletions

diff --git a/archaeological_finds/models_treatments.py b/archaeological_finds/models_treatments.py
index c35d3710c..512fd8a05 100644
--- a/archaeological_finds/models_treatments.py
+++ b/archaeological_finds/models_treatments.py
@@ -38,7 +38,7 @@ from ishtar_common.models import Document, GeneralType, \
     ImageModel, BaseHistorizedItem, OwnPerms, HistoricalRecords, Person, \
     Organization, ValueGetter, post_save_cache, ShortMenuItem, \
     DashboardFormItem, document_attached_changed, \
-    HistoryModel, SearchAltName, SearchVectorConfig
+    HistoryModel, SearchAltName, SearchVectorConfig, DocumentItem
 from ishtar_common.utils import cached_label_changed, get_current_year, \
     update_data, m2m_historization_changed
 
@@ -64,8 +64,8 @@ post_save.connect(post_save_cache, sender=TreatmentState)
 post_delete.connect(post_save_cache, sender=TreatmentState)
 
 
-class Treatment(DashboardFormItem, ValueGetter, BaseHistorizedItem,
-                ImageModel, OwnPerms, ShortMenuItem):
+class Treatment(DashboardFormItem, ValueGetter, DocumentItem,
+                BaseHistorizedItem, ImageModel, OwnPerms, ShortMenuItem):
     SLUG = 'treatment'
     SHOW_URL = 'show-treatment'
     TABLE_COLS = ('year', 'index', 'treatment_types__label',
@@ -872,8 +872,8 @@ post_save.connect(post_save_cache, sender=TreatmentFileType)
 post_delete.connect(post_save_cache, sender=TreatmentFileType)
 
 
-class TreatmentFile(DashboardFormItem, ClosedItem, BaseHistorizedItem,
-                    OwnPerms, ValueGetter, ShortMenuItem):
+class TreatmentFile(DashboardFormItem, ClosedItem, DocumentItem,
+                    BaseHistorizedItem, OwnPerms, ValueGetter, ShortMenuItem):
     SLUG = 'treatmentfile'
     SHOW_URL = 'show-treatmentfile'
     TABLE_COLS = ['type', 'year', 'index', 'internal_reference', 'name']