test: ok now?

1 files changed, 5 insertions, 4 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index 569f81669..1c30f57b0 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -63,9 +63,10 @@ class FindInit(ContextRecordInit):
         return self.finds, self.base_finds
 
     def get_default_find(self, force=False):
+        finds, base_finds = self.create_finds(force=force)
         if force:
-            return self.create_finds()[-1]
-        return self.create_finds()[0]
+            return finds[-1], base_finds[-1]
+        return finds[0], base_finds[0]
 
     def tearDown(self):
         super(FindInit, self).tearDown()
@@ -119,7 +120,7 @@ class AFindWizardCreationTest(WizardTest, FindInit, TestCase):
     ]
 
     def pre_wizard(self):
-        cr = self.create_context_record()[-1]
+        cr = self.get_default_context_record(force=True)
         self.form_datas[0].form_datas['selecrecord-find_creation']['pk'] = cr.pk
         self.find_number = models.Find.objects.count()
         self.basefind_number = models.BaseFind.objects.count()
@@ -184,7 +185,7 @@ class ATreatmentWizardCreationTest(WizardTest, FindInit, TestCase):
             person = Person.objects.create(name="default")
             person.id = 1
             person.save()
-        find = self.get_default_find(force=True)
+        find, base_find = self.get_default_find(force=True)
         self.form_datas[0].form_datas['selecfind-treatment_creation'][
             'pk'] = find.pk
         self.form_datas[0].form_datas['selecfind-treatment_creation'][