From 033da3ca632f58a9a14fc1f71daa113388e8f163 Mon Sep 17 00:00:00 2001
From: evanchooly <jlee@antwerkz.com>
Date: Wed, 22 Jan 2025 22:52:00 -0500
Subject: [PATCH] replace Datastore uses

---
 .../test/java/dev/morphia/test/mapping/TestReferences.java | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/core/src/test/java/dev/morphia/test/mapping/TestReferences.java b/core/src/test/java/dev/morphia/test/mapping/TestReferences.java
index e9d49d4c9aa..3dfd56de212 100644
--- a/core/src/test/java/dev/morphia/test/mapping/TestReferences.java
+++ b/core/src/test/java/dev/morphia/test/mapping/TestReferences.java
@@ -17,6 +17,7 @@
 import com.mongodb.client.model.ReturnDocument;
 
 import dev.morphia.Datastore;
+import dev.morphia.DatastoreImpl;
 import dev.morphia.Key;
 import dev.morphia.ModifyOptions;
 import dev.morphia.aggregation.Aggregation;
@@ -315,7 +316,7 @@ public void testInQueryAgainstReferences() {
 
     @Test
     public void testLazyWithParent() {
-        Datastore datastore = getDs();
+        DatastoreImpl datastore = getDs();
         datastore.getMapper().map(Entity1.class, Entity2.class, EntityBase.class);
 
         Entity1 entity1 = new Entity1("entity1");
@@ -403,8 +404,8 @@ public final void testMultiDimArrayPersistence() {
 
     @Test
     public void testMultipleDatabasesSingleThreaded() {
-        final Datastore ds1 = createDatastore(getMongoClient(), "db1");
-        final Datastore ds2 = createDatastore(getMongoClient(), "db2");
+        final DatastoreImpl ds1 = (DatastoreImpl) createDatastore(getMongoClient(), "db1");
+        final DatastoreImpl ds2 = (DatastoreImpl) createDatastore(getMongoClient(), "db2");
 
         final FacebookUser db1Friend = new FacebookUser(3, "DB1 FaceBook Friend");
         ds1.save(db1Friend);