diff --git a/main.py b/main.py
index ee18a68..4c21891 100644
--- a/main.py
+++ b/main.py
@@ -8,6 +8,7 @@ from nostr.event import Event, EventKind
 from nostr.relay_manager import RelayManager
 from nostr.key import PrivateKey
 from nostr.filter import Filter, Filters
+from nostr.message_type import ClientMessageType
 
 def get_relays():
     env_relays = os.getenv('RELAYS')