1 files changed, 8 insertions, 1 deletions
diff --git a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/identityprovider/api/SignedIdentityDocument.java b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/identityprovider/api/SignedIdentityDocument.java
index 6372c2202f0..60be42544c7 100644
--- a/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/identityprovider/api/SignedIdentityDocument.java
+++ b/vespa-athenz/src/main/java/com/yahoo/vespa/athenz/identityprovider/api/SignedIdentityDocument.java
@@ -28,6 +28,7 @@ public class SignedIdentityDocument {
     private final String instanceHostname;
     private final Instant createdAt;
     private final Set<String> ipAddresses;
+    private final IdentityType identityType;
 
     public SignedIdentityDocument(IdentityDocument identityDocument,
                                   String signature,
@@ -40,7 +41,8 @@ public class SignedIdentityDocument {
                                   String configServerHostname,
                                   String instanceHostname,
                                   Instant createdAt,
-                                  Set<String> ipAddresses) {
+                                  Set<String> ipAddresses,
+                                  IdentityType identityType) {
         this.identityDocument = identityDocument;
         this.signature = signature;
         this.signingKeyVersion = signingKeyVersion;
@@ -53,6 +55,7 @@ public class SignedIdentityDocument {
         this.instanceHostname = instanceHostname;
         this.createdAt = createdAt;
         this.ipAddresses = ipAddresses;
+        this.identityType = identityType;
     }
 
     public IdentityDocument identityDocument() {
@@ -102,4 +105,8 @@ public class SignedIdentityDocument {
     public Set<String> ipAddresses() {
         return ipAddresses;
     }
+
+    public IdentityType identityType() {
+        return identityType;
+    }
 }