address review comments

canonical · Apr 3, 2023 · 6543f98 · 6543f98
1 parent bab579a
commit 6543f98
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/lib/charms/kratos/v0/kratos_endpoints.py b/lib/charms/kratos/v0/kratos_endpoints.py
@@ -141,13 +141,7 @@ def get_kratos_endpoints(self) -> Optional[Dict]:
         if len(endpoints) == 0:
             raise KratosEndpointsRelationMissingError()
 
-        remote_app = [
-            app
-            for app in endpoints[0].data.keys()
-            if isinstance(app, Application) and not app._is_our_app
-        ][0]
-
-        data = endpoints[0].data[remote_app]
+        data = endpoints[0].data[endpoints[0].app]
 
         if "public_endpoint" not in data:
             raise KratosEndpointsRelationDataMissingError(