1 files changed, 9 insertions, 1 deletions
diff --git a/archaeological_warehouse/wizards.py b/archaeological_warehouse/wizards.py
index 6678e6c36..41c5c5945 100644
--- a/archaeological_warehouse/wizards.py
+++ b/archaeological_warehouse/wizards.py
@@ -20,13 +20,21 @@
 from django.shortcuts import render
 
 from ishtar_common.forms import reverse_lazy
-from ishtar_common.wizards import Wizard, DeletionWizard
+from ishtar_common.wizards import Wizard, DeletionWizard, SearchWizard
 from archaeological_finds.wizards import TreatmentWizard
 
 from archaeological_finds.models import Treatment, TreatmentType
 import models
 
 
+class WarehouseSearch(SearchWizard):
+    model = models.Warehouse
+
+
+class ContainerSearch(SearchWizard):
+    model = models.Container
+
+
 class PackagingWizard(TreatmentWizard):
     basket_step = 'base-packaging'