diff --git a/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj b/src/Tests.AzureStorage2/Tests.AzureStorage2.csproj
index 3fcc273..418c80f 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.1" />
   </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 c03a480..d841eb3 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.1" />
   </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 8c5b375..9a62ae8 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.1" />
   </ItemGroup>
 
   <ItemGroup Label="Direct references to transitive dependencies to avoid versions with CVE">