diff --git a/nautobot_ssot/integrations/aci/diffsync/models/nautobot.py b/nautobot_ssot/integrations/aci/diffsync/models/nautobot.py
index 1a945cc7d..1c084936f 100644
--- a/nautobot_ssot/integrations/aci/diffsync/models/nautobot.py
+++ b/nautobot_ssot/integrations/aci/diffsync/models/nautobot.py
@@ -485,5 +485,5 @@ def delete(self):
         return self
 
 
-NautobotDevice.update_forward_refs()
-NautobotDeviceType.update_forward_refs()
+NautobotDevice.model_rebuild()
+NautobotDeviceType.model_rebuild()
diff --git a/nautobot_ssot/integrations/dna_center/diffsync/models/base.py b/nautobot_ssot/integrations/dna_center/diffsync/models/base.py
index 18888b4bb..b4c81327f 100644
--- a/nautobot_ssot/integrations/dna_center/diffsync/models/base.py
+++ b/nautobot_ssot/integrations/dna_center/diffsync/models/base.py
@@ -158,6 +158,6 @@ class IPAddressOnInterface(DiffSyncModel):
     uuid: Optional[UUID] = None
 
 
-Area.update_forward_refs()
-Building.update_forward_refs()
-Device.update_forward_refs()
+Area.model_rebuild()
+Building.model_rebuild()
+Device.model_rebuild()