diff --git a/SqlDatabaseVectorSearch/Services/VectorSearchService.cs b/SqlDatabaseVectorSearch/Services/VectorSearchService.cs
index c4c44e1..32bc9ad 100644
--- a/SqlDatabaseVectorSearch/Services/VectorSearchService.cs
+++ b/SqlDatabaseVectorSearch/Services/VectorSearchService.cs
@@ -1,6 +1,5 @@
 using System.Data;
 using System.Runtime.CompilerServices;
-using System.Threading;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Options;
 using Microsoft.SemanticKernel.Embeddings;
@@ -28,7 +27,7 @@ public class VectorSearchService(IServiceProvider serviceProvider, ApplicationDb
         var strategy = dbContext.Database.CreateExecutionStrategy();
         var document = await strategy.ExecuteAsync(async (cancellationToken) =>
         {
-            await dbContext.Database.BeginTransactionAsync();
+            await dbContext.Database.BeginTransactionAsync(cancellationToken);
 
             if (documentId.HasValue)
             {
diff --git a/SqlDatabaseVectorSearch/SqlDatabaseVectorSearch.csproj b/SqlDatabaseVectorSearch/SqlDatabaseVectorSearch.csproj
index 842bfd3..a0e1b7b 100644
--- a/SqlDatabaseVectorSearch/SqlDatabaseVectorSearch.csproj
+++ b/SqlDatabaseVectorSearch/SqlDatabaseVectorSearch.csproj
@@ -18,7 +18,7 @@
         <PackageReference Include="Microsoft.ML.Tokenizers" Version="1.0.1" />
         <PackageReference Include="Microsoft.ML.Tokenizers.Data.Cl100kBase" Version="1.0.1" />
         <PackageReference Include="Microsoft.ML.Tokenizers.Data.O200kBase" Version="1.0.1" />
-        <PackageReference Include="Microsoft.SemanticKernel" Version="1.36.0" />
+        <PackageReference Include="Microsoft.SemanticKernel" Version="1.36.1" />
         <PackageReference Include="MinimalHelpers.OpenApi" Version="2.1.3" />
         <PackageReference Include="PdfPig" Version="0.1.9" />
         <PackageReference Include="Swashbuckle.AspNetCore.SwaggerUI" Version="7.2.0" />