diff --git a/client/src/main/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParameters.java b/client/src/main/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParameters.java
index 42a3d37f..6f90a4d0 100644
--- a/client/src/main/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParameters.java
+++ b/client/src/main/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParameters.java
@@ -2,6 +2,7 @@
 
 import com.sinch.sdk.core.models.OptionalValue;
 import com.sinch.sdk.domains.verification.models.Identity;
+import com.sinch.sdk.domains.verification.models.NumberIdentity;
 import com.sinch.sdk.domains.verification.models.VerificationMethodType;
 import com.sinch.sdk.domains.verification.models.VerificationReference;
 
@@ -83,7 +84,7 @@ protected Builder(VerificationMethodType method) {
      *     endpoint. number is currently the only supported endpoint type
      * @return current builder
      */
-    public B setIdentity(Identity identity) {
+    public B setIdentity(NumberIdentity identity) {
       this.identity = OptionalValue.of(identity);
       return self();
     }
diff --git a/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationFlashCallRequestParametersTest.java b/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationFlashCallRequestParametersTest.java
index aa9fa68c..9c8505fd 100644
--- a/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationFlashCallRequestParametersTest.java
+++ b/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationFlashCallRequestParametersTest.java
@@ -1,6 +1,5 @@
 package com.sinch.sdk.domains.verification.models.requests;
 
-import com.sinch.sdk.domains.verification.models.Identity;
 import com.sinch.sdk.domains.verification.models.NumberIdentity;
 import com.sinch.sdk.domains.verification.models.VerificationMethodType;
 import com.sinch.sdk.domains.verification.models.VerificationReference;
@@ -9,7 +8,7 @@
 
 class StartVerificationFlashCallRequestParametersTest {
 
-  final Identity identity = NumberIdentity.builder().setEndpoint("foo identity").build();
+  final NumberIdentity identity = NumberIdentity.builder().setEndpoint("foo identity").build();
   final VerificationMethodType method = VerificationMethodType.FLASH_CALL;
   final String reference = "foo reference";
   final String custom = "foo custom";
diff --git a/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParametersTest.java b/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParametersTest.java
index f4eb48f8..fa9cc978 100644
--- a/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParametersTest.java
+++ b/client/src/test/java/com/sinch/sdk/domains/verification/models/requests/StartVerificationRequestParametersTest.java
@@ -1,6 +1,5 @@
 package com.sinch.sdk.domains.verification.models.requests;
 
-import com.sinch.sdk.domains.verification.models.Identity;
 import com.sinch.sdk.domains.verification.models.NumberIdentity;
 import com.sinch.sdk.domains.verification.models.VerificationMethodType;
 import com.sinch.sdk.domains.verification.models.VerificationReference;
@@ -9,7 +8,7 @@
 
 class StartVerificationRequestParametersTest {
 
-  final Identity identity = NumberIdentity.builder().setEndpoint("foo identity").build();
+  final NumberIdentity identity = NumberIdentity.builder().setEndpoint("foo identity").build();
   final VerificationMethodType method = VerificationMethodType.CALLOUT;
   final String reference = "foo reference";
   final String custom = "foo custom";