1 files changed, 6 insertions, 3 deletions
diff --git a/synapse/rest/client/v2_alpha/register.py b/synapse/rest/client/v2_alpha/register.py
index 739c06eaca..91c0d5e981 100644
--- a/synapse/rest/client/v2_alpha/register.py
+++ b/synapse/rest/client/v2_alpha/register.py
@@ -672,16 +672,19 @@ class RegisterRestServlet(RestServlet):
             })
         defer.returnValue(result)
 
+    @defer.inlineCallbacks
     def _check_device_registered(self, user_id, device_id, initial_display_name):
 
         if self.hs.config.worker_app:
-            return self._device_check_registered_client(
+            r = yield self._device_check_registered_client(
                 user_id, device_id, initial_display_name
-            )["device_id"]
+            )
+            defer.returnValue(r["device_id"])
         else:
-            return self.device_handler.check_device_registered(
+            r = yield self.device_handler.check_device_registered(
                 user_id, device_id, initial_display_name
             )
+            defer.returnValue(r)
 
     def _register_device(self, user_id, params):
         """Register a device for a user.