fix strip_invalid_mxid_characters

1 files changed, 2 insertions, 2 deletions

diff --git a/synapse/rest/client/v2_alpha/register.py b/synapse/rest/client/v2_alpha/register.py

index b5704a39d9..36676c9fa8 100644
--- a/synapse/rest/client/v2_alpha/register.py
+++ b/synapse/rest/client/v2_alpha/register.py
@@ -387,7 +387,7 @@ class RegisterRestServlet(RestServlet):
                     self.hs.config.register_mxid_from_3pid == 'email' and
                     LoginType.EMAIL_IDENTITY in auth_result
                 ):
-                    address = auth_result[login_type]['address']
+                    address = auth_result[LoginType.EMAIL_IDENTITY]['address']
                     desired_username = types.strip_invalid_mxid_characters(
                         address.replace('@', '-').lower()
                     )
@@ -427,7 +427,7 @@ class RegisterRestServlet(RestServlet):
                     self.hs.config.register_mxid_from_3pid == 'msisdn' and
                     LoginType.MSISDN in auth_result
                 ):
-                    desired_username = auth_result[login_type]['address']
+                    desired_username = auth_result[LoginType.MSISDN]['address']
                 else:
                     raise SynapseError(
                         400, "Cannot derive mxid from 3pid; no recognised 3pid"