diff --git a/v3/src/main/java/com/skyflow/utils/Utils.java b/v3/src/main/java/com/skyflow/utils/Utils.java
index e44395db..76edb9cd 100644
--- a/v3/src/main/java/com/skyflow/utils/Utils.java
+++ b/v3/src/main/java/com/skyflow/utils/Utils.java
@@ -125,7 +125,7 @@ public static com.skyflow.vault.data.InsertResponse formatResponse(InsertRespons
                             tokensMap.put(key, tokenList);
                         }
                     }
-                    Success success = new Success(index, record.get(index).getSkyflowId().get(), tokensMap, null);
+                    Success success = new Success(indexNumber, record.get(index).getSkyflowId().get(), tokensMap, null);
                     successRecords.add(success);
                 }
                 indexNumber++;
diff --git a/v3/src/main/java/com/skyflow/vault/controller/VaultController.java b/v3/src/main/java/com/skyflow/vault/controller/VaultController.java
index 08d38aba..40093fb0 100644
--- a/v3/src/main/java/com/skyflow/vault/controller/VaultController.java
+++ b/v3/src/main/java/com/skyflow/vault/controller/VaultController.java
@@ -219,4 +219,4 @@ private void configureInsertConcurrencyAndBatchSize(int totalRequests) {
             this.insertConcurrencyLimit = Math.min(Constants.INSERT_CONCURRENCY_LIMIT, maxConcurrencyNeeded);
         }
     }
-}
+}
\ No newline at end of file