1 files changed, 4 insertions, 2 deletions
diff --git a/Firestore/Source/API/FIRFirestore+Internal.h b/Firestore/Source/API/FIRFirestore+Internal.h
index c2e995a..717a08b 100644
--- a/Firestore/Source/API/FIRFirestore+Internal.h
+++ b/Firestore/Source/API/FIRFirestore+Internal.h
@@ -16,9 +16,10 @@
 
 #import "FIRFirestore.h"
 
+#include "Firestore/core/src/firebase/firestore/model/database_id.h"
+
 NS_ASSUME_NONNULL_BEGIN
 
-@class FSTDatabaseID;
 @class FSTDispatchQueue;
 @class FSTFirestoreClient;
 @class FSTUserDataConverter;
@@ -54,7 +55,8 @@ NS_ASSUME_NONNULL_BEGIN
 - (void)shutdownWithCompletion:(nullable void (^)(NSError *_Nullable error))completion
     NS_SWIFT_NAME(shutdown(completion:));
 
-@property(nonatomic, strong, readonly) FSTDatabaseID *databaseID;
+// FIRFirestore ownes the DatabaseId instance.
+@property(nonatomic, assign, readonly) const firebase::firestore::model::DatabaseId *databaseID;
 @property(nonatomic, strong, readonly) FSTFirestoreClient *client;
 @property(nonatomic, strong, readonly) FSTUserDataConverter *dataConverter;