1 files changed, 13 insertions, 1 deletions
diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py
index b15948dce..6486e5da1 100644
--- a/archaeological_finds/wizards.py
+++ b/archaeological_finds/wizards.py
@@ -21,7 +21,7 @@ from django.core.exceptions import ObjectDoesNotExist, PermissionDenied
 from django.utils.translation import ugettext_lazy as _
 
 from ishtar_common.forms import reverse_lazy
-from ishtar_common.wizards import Wizard, DeletionWizard, SourceWizard
+from ishtar_common.wizards import Wizard, DeletionWizard, SearchWizard
 from archaeological_operations.wizards import OperationAdministrativeActWizard
 
 from archaeological_operations.models import AdministrativeAct
@@ -29,6 +29,10 @@ from archaeological_context_records.models import ContextRecord
 import models
 
 
+class FindSearch(SearchWizard):
+    model = models.Find
+
+
 class FindWizard(Wizard):
     model = models.Find
     wizard_done_window = reverse_lazy('show-find')
@@ -98,6 +102,10 @@ class FindDeletionWizard(DeletionWizard):
               'comment']
 
 
+class TreatmentSearch(SearchWizard):
+    model = models.Treatment
+
+
 class TreatmentWizard(Wizard):
     model = models.Treatment
     wizard_done_window = reverse_lazy('show-treatment')
@@ -162,6 +170,10 @@ class TreatmentEditAdministrativeActWizard(TreatmentAdministrativeActWizard):
         return self.get_current_object().treatment
 
 
+class TreatmentFileSearch(SearchWizard):
+    model = models.TreatmentFile
+
+
 class TreatmentFileWizard(Wizard):
     model = models.TreatmentFile
     wizard_done_window = reverse_lazy('show-treatmentfile')