diff --git a/python/src/uagents/registration.py b/python/src/uagents/registration.py
index 41b136c5..4c6cd2b5 100644
--- a/python/src/uagents/registration.py
+++ b/python/src/uagents/registration.py
@@ -20,6 +20,7 @@
 )
 from uagents.crypto import Identity
 from uagents.network import AlmanacContract, InsufficientFundsError, add_testnet_funds
+from uagents.types import AgentEndpoint
 
 
 class AgentRegistrationPolicy(ABC):