Making DocumentSnapshot nullable

1 files changed, 7 insertions, 7 deletions

diff --git a/Firestore/Source/API/FIRQuerySnapshot.m b/Firestore/Source/API/FIRQuerySnapshot.m
index 6bc6761..5e1af9a 100644
--- a/Firestore/Source/API/FIRQuerySnapshot.m
+++ b/Firestore/Source/API/FIRQuerySnapshot.m
@@ -57,7 +57,7 @@ NS_ASSUME_NONNULL_BEGIN
 
 @implementation FIRQuerySnapshot {
   // Cached value of the documents property.
-  NSArray<FIRDocumentSnapshot *> *_documents;
+  NSArray<FIRQueryDocumentSnapshot *> *_documents;
 
   // Cached value of the documentChanges property.
   NSArray<FIRDocumentChange *> *_documentChanges;
@@ -93,18 +93,18 @@ NS_ASSUME_NONNULL_BEGIN
   return self.snapshot.documents.count;
 }
 
-- (NSArray<FIRDocumentSnapshot *> *)documents {
+- (NSArray<FIRQueryDocumentSnapshot *> *)documents {
   if (!_documents) {
     FSTDocumentSet *documentSet = self.snapshot.documents;
     FIRFirestore *firestore = self.firestore;
     BOOL fromCache = self.metadata.fromCache;
 
-    NSMutableArray<FIRDocumentSnapshot *> *result = [NSMutableArray array];
+    NSMutableArray<FIRQueryDocumentSnapshot *> *result = [NSMutableArray array];
     for (FSTDocument *document in documentSet.documentEnumerator) {
-      [result addObject:[FIRDocumentSnapshot snapshotWithFirestore:firestore
-                                                       documentKey:document.key
-                                                          document:document
-                                                         fromCache:fromCache]];
+      [result addObject:[FIRQueryDocumentSnapshot snapshotWithFirestore:firestore
+                                                            documentKey:document.key
+                                                               document:document
+                                                              fromCache:fromCache]];
     }
 
     _documents = result;