diff --git a/fabric-data-generation-api-v1/src/main/java/net/fabricmc/fabric/impl/datagen/loot/FabricLootTableProviderImpl.java b/fabric-data-generation-api-v1/src/main/java/net/fabricmc/fabric/impl/datagen/loot/FabricLootTableProviderImpl.java
index 9b77fe69ed4..2e4f38db00b 100644
--- a/fabric-data-generation-api-v1/src/main/java/net/fabricmc/fabric/impl/datagen/loot/FabricLootTableProviderImpl.java
+++ b/fabric-data-generation-api-v1/src/main/java/net/fabricmc/fabric/impl/datagen/loot/FabricLootTableProviderImpl.java
@@ -58,7 +58,7 @@ public static CompletableFuture<?> run(
 		HashMap<Identifier, LootTable> builders = Maps.newHashMap();
 		HashMap<Identifier, ConditionJsonProvider[]> conditionMap = new HashMap<>();
 
-		return registryLookup.thenApply(lookup -> {
+		return registryLookup.thenCompose(lookup -> {
 			provider.accept(lookup, (identifier, builder) -> {
 				ConditionJsonProvider[] conditions = FabricDataGenHelper.consumeConditions(builder);
 				conditionMap.put(identifier, conditions);