diff --git a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/IdPhotoResponse.java b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/IdPhotoResponse.java
index 79c27f8f..ca0a394e 100644
--- a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/IdPhotoResponse.java
+++ b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/IdPhotoResponse.java
@@ -1,16 +1,14 @@
 package com.yoti.api.client.docs.session.retrieve;
 
-import com.yoti.api.client.Media;
-
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 public class IdPhotoResponse {
 
     @JsonProperty("media")
-    private Media media;
+    private MediaResponse media;
 
-    public Media getMedia() {
+    public MediaResponse getMedia() {
         return media;
     }
-    
+
 }
diff --git a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/LookupProfileResponse.java b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/LookupProfileResponse.java
index 8b774239..450904b1 100644
--- a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/LookupProfileResponse.java
+++ b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/LookupProfileResponse.java
@@ -1,15 +1,13 @@
 package com.yoti.api.client.docs.session.retrieve;
 
-import com.yoti.api.client.Media;
-
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 public class LookupProfileResponse {
 
     @JsonProperty("media")
-    private Media media;
+    private MediaResponse media;
 
-    public Media getMedia() {
+    public MediaResponse getMedia() {
         return media;
     }
 
diff --git a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/ReturnedProfileResponse.java b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/ReturnedProfileResponse.java
index 486f7e05..ad2b9035 100644
--- a/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/ReturnedProfileResponse.java
+++ b/yoti-sdk-api/src/main/java/com/yoti/api/client/docs/session/retrieve/ReturnedProfileResponse.java
@@ -1,16 +1,14 @@
 package com.yoti.api.client.docs.session.retrieve;
 
-import com.yoti.api.client.Media;
-
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 public class ReturnedProfileResponse {
 
     @JsonProperty("media")
-    private Media media;
+    private MediaResponse media;
 
-    public Media getMedia() {
+    public MediaResponse getMedia() {
         return media;
     }
-    
+
 }