From b760d45c8bea5757d29cad5754a116adfd927624 Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Mon, 27 Oct 2025 07:59:59 -0700
Subject: [PATCH] fix test

---
 tests/app/organization/test_rest.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/tests/app/organization/test_rest.py b/tests/app/organization/test_rest.py
index 698fbe3c3..348ab4e46 100644
--- a/tests/app/organization/test_rest.py
+++ b/tests/app/organization/test_rest.py
@@ -960,10 +960,15 @@ def test_get_organization_message_allowance(admin_request, sample_organization,
     assert response["total_message_limit"] == 150000
 
     assert mock_get_counts.call_count == 1
-    mock_get_counts.assert_called_once_with([service_1.id, service_2.id], 2025)
+    mock_get_counts.assert_called_once()
+    args, _ = mock_get_counts.call_args
+    assert set(args[0]) == {service_1.id, service_2.id}
+    assert args[1] == 2025
 
 
-def test_get_organization_message_allowance_no_services(admin_request, sample_organization):
+def test_get_organization_message_allowance_no_services(
+    admin_request, sample_organization
+):
     response = admin_request.get(
         "organization.get_organization_message_allowance",
         organization_id=sample_organization.id,