diff --git a/opentech/apply/funds/management/commands/migration_base.py b/opentech/apply/funds/management/commands/migration_base.py index 1472e2df380f80b7be557862070c9247ce291b74..3014aef133d68a26a3d0a050969d1c9c055d37e4 100644 --- a/opentech/apply/funds/management/commands/migration_base.py +++ b/opentech/apply/funds/management/commands/migration_base.py @@ -58,7 +58,12 @@ class MigrateCommand(BaseCommand): submission.page = FUND ROUND = Round.objects.get(title=self.ROUND_NAME) submission.round = ROUND - FORM = RoundForm.objects.get(round=ROUND) + if self.APPLICATION_TYPE == "request": + FORM = RoundForm.objects.get(round=ROUND) + elif self.APPLICATION_TYPE == "concept": + FORM = RoundForm.objects.filter(round=ROUND)[0] + elif self.APPLICATION_TYPE == "proposal": + FORM = RoundForm.objects.filter(round=ROUND)[1] submission.form_fields = FORM.form.form_fields elif self.CONTENT_TYPE == "lab": LAB = LabType.objects.get(title=self.LAB_NAME)