diff --git a/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj b/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj
index f9df460..abaaded 100644
--- a/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj
+++ b/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj
@@ -24,7 +24,7 @@
   <ItemGroup Label="These dependencies must reference NServiceBus 7 versions">
     <PackageReference Include="NServiceBus.AcceptanceTesting" Version="7.8.6" />
     <PackageReference Include="NServiceBus.Persistence.AzureStorage" Version="2.4.2" />
-    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="1.2.2" />
+    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="4.0.0" />
   </ItemGroup>
 
   <ItemGroup Label="Direct references to transitive dependencies to avoid versions with CVE">
diff --git a/src/Tests.AzureTable3/Tests.AzureTable3.csproj b/src/Tests.AzureTable3/Tests.AzureTable3.csproj
index 5b66c6f..d8d21f1 100644
--- a/src/Tests.AzureTable3/Tests.AzureTable3.csproj
+++ b/src/Tests.AzureTable3/Tests.AzureTable3.csproj
@@ -24,7 +24,7 @@ we can't update the library because the tests here need to reference an older ve
   <ItemGroup Label="These dependencies must reference NServiceBus 7 versions">
     <PackageReference Include="NServiceBus.AcceptanceTesting" Version="7.8.6" />
     <PackageReference Include="NServiceBus.Persistence.AzureTable" Version="3.2.2" />
-    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="1.2.2" />
+    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="4.0.0" />
   </ItemGroup>
 
   <ItemGroup Label="Direct references to transitive dependencies to avoid versions with CVE">
diff --git a/src/Tests.AzureTable4/Tests.AzureTable4.csproj b/src/Tests.AzureTable4/Tests.AzureTable4.csproj
index 20a4392..9cee40d 100644
--- a/src/Tests.AzureTable4/Tests.AzureTable4.csproj
+++ b/src/Tests.AzureTable4/Tests.AzureTable4.csproj
@@ -24,7 +24,7 @@ we can't update the library because the tests here need to reference an older ve
   <ItemGroup Label="These dependencies must reference NServiceBus 8 versionss">
     <PackageReference Include="NServiceBus.AcceptanceTesting" Version="8.2.4" />
     <PackageReference Include="NServiceBus.Persistence.AzureTable" Version="4.0.1" />
-    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="2.0.1" />
+    <PackageReference Include="NServiceBus.Persistence.CosmosDB" Version="4.0.0" />
   </ItemGroup>
 
   <ItemGroup Label="Direct references to transitive dependencies to avoid versions with CVE">