From 55d346fe48464eb5ae0adc0fd7d013037338275c Mon Sep 17 00:00:00 2001
From: ameerabdul <51499575+ameerabdul@users.noreply.github.com>
Date: Mon, 10 Mar 2025 12:08:58 -0500
Subject: [PATCH] fix: Dataloader helper Try set values

---
 src/main/java/org/dataloader/DataLoaderHelper.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/dataloader/DataLoaderHelper.java b/src/main/java/org/dataloader/DataLoaderHelper.java
index 9b5a59e..8245284 100644
--- a/src/main/java/org/dataloader/DataLoaderHelper.java
+++ b/src/main/java/org/dataloader/DataLoaderHelper.java
@@ -429,7 +429,11 @@ CompletableFuture<List<V>> invokeLoader(List<K> keys, List<Object> keyContexts,
                     for (int i = 0; i < missedValues.size(); i++) {
                         V v = missedValues.get(i);
                         Integer listIndex = missedKeyIndexes.get(i);
-                        valuesInKeyOrder.set(listIndex, Try.succeeded(v));
+                        if (v instanceof Try) {
+                            valuesInKeyOrder.set(listIndex, (Try<V>) v);
+                        } else {
+                            valuesInKeyOrder.set(listIndex, Try.succeeded(v));
+                        }
                     }
                     List<V> assembledValues = valuesInKeyOrder.stream().map(Try::get).collect(toList());
                     //