diff --git a/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsDataLoaderProvider.kt b/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsDataLoaderProvider.kt
index d8740b244..2df4d2093 100644
--- a/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsDataLoaderProvider.kt
+++ b/graphql-dgs/src/main/kotlin/com/netflix/graphql/dgs/internal/DgsDataLoaderProvider.kt
@@ -183,7 +183,7 @@ class DgsDataLoaderProvider(private val applicationContext: ApplicationContext)
             extendedBatchLoader.setDataLoaderRegistry(dataLoaderRegistry)
         }
 
-        return DataLoaderFactory.newDataLoader(batchLoader, options)
+        return DataLoaderFactory.newDataLoader(extendedBatchLoader, options)
     }
 
     private fun <T> createDataLoader(