diff --git a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Query/QueryPartitionProviderTests.cs b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Query/QueryPartitionProviderTests.cs
index ccd7a81dfd..e4e6f48846 100644
--- a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Query/QueryPartitionProviderTests.cs
+++ b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/Query/QueryPartitionProviderTests.cs
@@ -39,7 +39,8 @@ public void TestQueryPartitionProviderUpdate()
                     allowNonValueAggregateQuery: true,
                     hasLogicalPartitionKey: false,
                     allowDCount: true,
-                    useSystemPrefix: false);
+                    useSystemPrefix: false,
+                    geospatialType: Cosmos.GeospatialType.Geography);
 
             Assert.IsTrue(tryGetQueryPlan.Failed);
             Assert.IsTrue(tryGetQueryPlan.Exception.ToString().Contains("The SQL query text exceeded the maximum limit of 5 characters"));
@@ -54,7 +55,8 @@ public void TestQueryPartitionProviderUpdate()
                             allowNonValueAggregateQuery: true,
                             hasLogicalPartitionKey: false,
                             allowDCount: true,
-                            useSystemPrefix: false);
+                            useSystemPrefix: false,
+                            geospatialType: Cosmos.GeospatialType.Geography);
 
             Assert.IsTrue(tryGetQueryPlan.Succeeded);
         }