diff --git a/synapse_invite_checker/invite_checker.py b/synapse_invite_checker/invite_checker.py
index 41f923a..41ff2df 100644
--- a/synapse_invite_checker/invite_checker.py
+++ b/synapse_invite_checker/invite_checker.py
@@ -1,7 +1,8 @@
+import requests
 import json
 import time
 import os
-import requests
+from twisted.internet.defer import inlineCallbacks, returnValue
 from synapse.module_api import ModuleApi, errors
 from synapse.types import UserID
 import logging