diff --git a/src/OrchardCore/OrchardCore.Infrastructure/Shell/ShellDescriptorManager.cs b/src/OrchardCore/OrchardCore.Infrastructure/Shell/ShellDescriptorManager.cs
index a6673ab3a2a..eb50f708ce1 100644
--- a/src/OrchardCore/OrchardCore.Infrastructure/Shell/ShellDescriptorManager.cs
+++ b/src/OrchardCore/OrchardCore.Infrastructure/Shell/ShellDescriptorManager.cs
@@ -159,8 +159,8 @@ private async Task UpgradeFromBeta2()
 
                     var updateShellStateCmd = $"UPDATE {documentTable} SET {dialect.QuoteForColumnName(nameof(Document.Type))} = {dialect.GetSqlValue(newShellStateType)} WHERE {dialect.QuoteForColumnName(nameof(Document.Type))} = {dialect.GetSqlValue(oldShellStateType)}";
 
-                    await connection.ExecuteAsync(updateShellDescriptorCmd);
-                    await connection.ExecuteAsync(updateShellStateCmd);
+                    await connection.ExecuteAsync(updateShellDescriptorCmd, null, transaction);
+                    await connection.ExecuteAsync(updateShellStateCmd, null, transaction);
 
                     transaction.Commit();
                 }