pythonmobile
diff --git a/‎openid/consumer/consumer.py‎
Lines changed: 7 additions & 5 deletions b/‎openid/consumer/consumer.py‎
Lines changed: 7 additions & 5 deletions
@@ -1689,6 +1689,13 @@ def setEndpoint(self, endpoint):
         else:
             self.identity_url = endpoint.claimed_id
 
+    def getDisplayIdentifier(self):
+        """Return the display identifier for this response.
+        """
+        if self.endpoint is not None:
+            return self.endpoint.getDisplayIdentifier()
+        return None
+
 class SuccessResponse(Response):
     """A response with a status of SUCCESS. Indicates that this request is a
     successful acknowledgement from the OpenID server that the
@@ -1721,11 +1728,6 @@ def __init__(self, endpoint, message, signed_fields=None):
             signed_fields = []
         self.signed_fields = signed_fields
 
-    def getDisplayIdentifier(self):
-        """Return the display identifier for this success response.
-        """
-        return self.endpoint.getDisplayIdentifier()
-
     def isOpenID1(self):
         """Was this authentication response an OpenID 1 authentication
         response?