change type :)

goshippo · Nov 19, 2024 · 6824c16 · 6824c16
1 parent 841aba5
commit 6824c16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_service_group.py b/tests/test_service_group.py
@@ -1,4 +1,4 @@
-from shippo.models.components import ServiceGroup, ServiceLevel
+from shippo.models.components import ServiceGroup, ServiceGroupAccountAndServiceLevel
 
 import shippo
 
@@ -22,6 +22,6 @@ def test_service_groups(self, api: shippo.Shippo):
 
         first_service_level = first_service_group.service_levels[0]
         assert first_service_level is not None
-        assert isinstance(first_service_level, ServiceLevel)
+        assert isinstance(first_service_level, ServiceGroupAccountAndServiceLevel)
         assert first_service_level.account_object_id is not None
         assert first_service_level.service_level_token is not None