diff --git a/src/OrchardCore.Modules/OrchardCore.Flows/Drivers/BagPartDisplayDriver.cs b/src/OrchardCore.Modules/OrchardCore.Flows/Drivers/BagPartDisplayDriver.cs
index 18019918baf..f2d677022af 100644
--- a/src/OrchardCore.Modules/OrchardCore.Flows/Drivers/BagPartDisplayDriver.cs
+++ b/src/OrchardCore.Modules/OrchardCore.Flows/Drivers/BagPartDisplayDriver.cs
@@ -79,6 +79,7 @@ public override IDisplayResult Edit(BagPart bagPart, BuildPartEditorContext cont
                 m.Updater = context.Updater;
                 m.ContainedContentTypeDefinitions = await GetContainedContentTypesAsync(context.TypePartDefinition);
                 m.AccessibleWidgets = await GetAccessibleWidgetsAsync(bagPart.ContentItems, contentDefinitionManager);
+                m.TypePartDefinition = context.TypePartDefinition;
             });
         }
 
diff --git a/src/OrchardCore.Modules/OrchardCore.Flows/ViewModels/BagPartEditViewModel.cs b/src/OrchardCore.Modules/OrchardCore.Flows/ViewModels/BagPartEditViewModel.cs
index 53c5c465652..f58f0713cbe 100644
--- a/src/OrchardCore.Modules/OrchardCore.Flows/ViewModels/BagPartEditViewModel.cs
+++ b/src/OrchardCore.Modules/OrchardCore.Flows/ViewModels/BagPartEditViewModel.cs
@@ -25,5 +25,8 @@ public class BagPartEditViewModel
 
         [BindNever]
         public IEnumerable<BagPartWidgetViewModel> AccessibleWidgets { get; set; }
+
+        [BindNever]
+        public ContentTypePartDefinition TypePartDefinition { get; set; }
     }
 }