diff --git a/src/EFCore.Relational/Query/RelationalShapedQueryCompilingExpressionVisitor.cs b/src/EFCore.Relational/Query/RelationalShapedQueryCompilingExpressionVisitor.cs
index e623a1e92e7..a41535e943a 100644
--- a/src/EFCore.Relational/Query/RelationalShapedQueryCompilingExpressionVisitor.cs
+++ b/src/EFCore.Relational/Query/RelationalShapedQueryCompilingExpressionVisitor.cs
@@ -200,7 +200,8 @@ private static Task<int> NonQueryResultAsync(
                                 state.commandSource),
                             cancellationToken);
                     },
-                    null);
+                    null,
+                    relationalQueryContext.CancellationToken);
             }
             finally
             {