diff --git a/src/Elasticsearch/State/CollectionProvider.php b/src/Elasticsearch/State/CollectionProvider.php
index df84a52e32d..1708a96a0f6 100644
--- a/src/Elasticsearch/State/CollectionProvider.php
+++ b/src/Elasticsearch/State/CollectionProvider.php
@@ -87,6 +87,8 @@ public function provide(Operation $operation, array $uriVariables = [], array $c
             $documents = $documents->asArray();
         }
 
+        $resourceClass = $operation->getOutput()['class'] ?? $resourceClass;
+
         return new Paginator(
             $this->denormalizer,
             $documents,