diff --git a/tests/test_service_group.py b/tests/test_service_group.py
index 8cfd835..d7e2fe9 100644
--- 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