diff --git a/backend/support/test_cog_over.py b/backend/support/test_cog_over.py index a5d4e929e..6dd482d2b 100644 --- a/backend/support/test_cog_over.py +++ b/backend/support/test_cog_over.py @@ -506,7 +506,7 @@ def test_cog_assignment_for_2024_audit(self): sac.save() cog_agency, over_agency = compute_cog_over( - sac.federal_awards, sac.submission_status, sac.general_information["ein"], + sac.federal_awards, sac.submission_status, sac.general_information["ein"], sac.general_information["auditee_uei"], sac.general_information["audit_year"] ) self.assertEqual(cog_agency, "84") @@ -545,7 +545,7 @@ def test_cog_assignment_for_2027_w_baseline(self): sac.save() cog_agency, over_agency = compute_cog_over( - sac.federal_awards, sac.submission_status, sac.general_information["ein"], + sac.federal_awards, sac.submission_status, sac.general_information["ein"], sac.general_information["auditee_uei"], sac.general_information["audit_year"] ) self.assertEqual(cog_agency, "84") @@ -584,7 +584,7 @@ def test_cog_assignment_for_2027_no_baseline(self): sac.save() cog_agency, over_agency = compute_cog_over( - sac.federal_awards, sac.submission_status, sac.general_information["ein"], + sac.federal_awards, sac.submission_status, sac.general_information["ein"], sac.general_information["auditee_uei"], sac.general_information["audit_year"] ) self.assertEqual(cog_agency, "10")