diff --git a/opentech/apply/funds/tests/test_admin_form.py b/opentech/apply/funds/tests/test_admin_form.py
index cafd805c2054695c01cf44a4b99a479d4be91417..93413519a472dc54e3a217bad6d76e9588e9c533 100644
--- a/opentech/apply/funds/tests/test_admin_form.py
+++ b/opentech/apply/funds/tests/test_admin_form.py
@@ -66,7 +66,7 @@ class TestWorkflowFormAdminForm(TestCase):
         self.assertTrue(form.is_valid(), form.errors.as_text())
 
     def test_doesnt_validates_with_two_forms_one_stage(self):
-        form = self.submit_data(form_data(2, stages=1))
+        form = self.submit_data(form_data(2))
         self.assertFalse(form.is_valid())
         self.assertTrue(form.errors['__all__'])
         formset_errors = form.formsets['forms'].errors
@@ -76,5 +76,5 @@ class TestWorkflowFormAdminForm(TestCase):
         self.assertTrue(formset_errors[1]['form'])
 
     def test_can_save_two_forms(self):
-        form = self.submit_data(form_data(2))
+        form = self.submit_data(form_data(2, stages=2))
         self.assertTrue(form.is_valid())
diff --git a/opentech/apply/funds/tests/test_admin_views.py b/opentech/apply/funds/tests/test_admin_views.py
index b065ad10da7a3acf82b850bb41ce1d33478af9da..3469b0448bbab7eb7bdf37c197628fa22d2ee2d4 100644
--- a/opentech/apply/funds/tests/test_admin_views.py
+++ b/opentech/apply/funds/tests/test_admin_views.py
@@ -17,7 +17,7 @@ class TestFundCreationView(TestCase):
         self.client.force_login(self.user)
         url = reverse('wagtailadmin_pages:add', args=('funds', 'fundtype', self.home.id))
 
-        data = form_data(forms, same_forms=same_forms)
+        data = form_data(forms, same_forms=same_forms, stages=forms)
         data['action-publish'] = True
 
         response = self.client.post(url, data=data, secure=True, follow=True)