diff --git a/toontown/uberdog/DistributedPartyManagerUD.py b/toontown/uberdog/DistributedPartyManagerUD.py
index f62594b..acff337 100644
--- a/toontown/uberdog/DistributedPartyManagerUD.py
+++ b/toontown/uberdog/DistributedPartyManagerUD.py
@@ -1,5 +1,5 @@
 from direct.directnotify import DirectNotifyGlobal
-from direct.distributed.DistributedObjectAI import DistributedObjectAI
+from direct.distributed.DistributedObjectUD import DistributedObjectUD
 
-class DistributedPartyManagerUD(DistributedObjectAI):
+class DistributedPartyManagerUD(DistributedObjectUD):
     notify = DirectNotifyGlobal.directNotify.newCategory('DistributedPartyManagerUD')