diff --git a/Lombiq.HelpfulLibraries.OrchardCore/Contents/JsonSectionDisplayDriver.cs b/Lombiq.HelpfulLibraries.OrchardCore/Contents/JsonSectionDisplayDriver.cs
index f6ed2002..810fb7d4 100644
--- a/Lombiq.HelpfulLibraries.OrchardCore/Contents/JsonSectionDisplayDriver.cs
+++ b/Lombiq.HelpfulLibraries.OrchardCore/Contents/JsonSectionDisplayDriver.cs
@@ -3,6 +3,7 @@
 using Microsoft.AspNetCore.Mvc.ModelBinding;
 using OrchardCore.DisplayManagement.Entities;
 using OrchardCore.DisplayManagement.Handlers;
+using OrchardCore.DisplayManagement.ModelBinding;
 using OrchardCore.DisplayManagement.Views;
 using OrchardCore.Security.Permissions;
 using OrchardCore.Settings;
@@ -43,7 +44,7 @@ await AuthorizeAsync()
                 .OnGroup(GroupId)
             : null;
 
-    public override async Task<IDisplayResult> UpdateAsync(TSection section, BuildEditorContext context)
+    public override async Task<IDisplayResult> UpdateAsync(TSection section, UpdateEditorContext context)
     {
         var viewModel = new JsonViewModel<TAdditionalData>();