diff --git a/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml b/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml
index fb4852f..e6f4b68 100644
--- a/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml
+++ b/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml
@@ -18,13 +18,13 @@
                             <ColumnDefinition Width="Auto"/>
                         </Grid.ColumnDefinitions>
                         <!-- ToolBox Control -->
-                        <ContentControl Template="{Binding ToolBox,RelativeSource={RelativeSource AncestorType={x:Type UserControl}}}"/>
+                        <ContentControl Template="{Binding ToolBox,RelativeSource={RelativeSource AncestorType={x:Type Control}}}"/>
 
                         <ScrollViewer Grid.Column="1">
                             <!-- Diagram Control -->
                             <dd:DiagramControl x:Name="PART_DiagramControl" MinWidth="1000" MinHeight="1000" HorizontalAlignment="Stretch" VerticalAlignment="Stretch" />
                         </ScrollViewer>
-                        <ContentControl Grid.Column="2" x:Name="properity" Template="{Binding PropertiesBox,RelativeSource={RelativeSource AncestorType={x:Type UserControl}}}"/>
+                        <ContentControl Grid.Column="2" x:Name="properity" Template="{Binding PropertiesBox,RelativeSource={RelativeSource AncestorType={x:Type Control}}}"/>
 
                     </Grid>
                 </ControlTemplate>
diff --git a/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml.cs b/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml.cs
index 832daeb..8c81be6 100644
--- a/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml.cs
+++ b/Extensions/AIStudio.Wpf.Flowchart/Controls/FlowchartEditor.xaml.cs
@@ -18,7 +18,7 @@ namespace AIStudio.Wpf.Flowchart.Controls
     /// FlowchartEditor.xaml 的交互逻辑
     /// </summary>
     [TemplatePart(Name = PART_DiagramControl, Type = typeof(DiagramControl))]
-    public partial class FlowchartEditor : UserControl
+    public partial class FlowchartEditor : Control
     {
         public const string PART_DiagramControl = "PART_DiagramControl";
         private DiagramControl _diagramControl;
diff --git a/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml b/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml
index 04556a1..567570e 100644
--- a/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml
+++ b/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml
@@ -21,7 +21,7 @@
                             <RowDefinition/>
                         </Grid.RowDefinitions>
                         <!-- ToolBox Control -->
-                        <ContentControl x:Name="PART_ContentControl" Template="{Binding ToolBox,RelativeSource={RelativeSource AncestorType={x:Type UserControl}}}"/>
+                        <ContentControl x:Name="PART_ContentControl" Template="{Binding ToolBox,RelativeSource={RelativeSource AncestorType={x:Type Control}}}"/>
 
                         <!-- Diagram Control -->
                         <dd:DiagramControl Grid.Row="1" x:Name="PART_DiagramControl"  HorizontalAlignment="Stretch" VerticalAlignment="Stretch" />
diff --git a/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml.cs b/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml.cs
index 39c7f63..dec24f0 100644
--- a/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml.cs
+++ b/Extensions/AIStudio.Wpf.Mind/Controls/MindEditor.xaml.cs
@@ -21,7 +21,7 @@ namespace AIStudio.Wpf.Mind.Controls
     [TemplatePart(Name = PART_DiagramControl, Type = typeof(DiagramControl))]
     [TemplatePart(Name = PART_ContentControl, Type = typeof(ContentControl))]
     [TemplatePart(Name = PART_SearchControl, Type = typeof(Border))]
-    public partial class MindEditor : UserControl
+    public partial class MindEditor : Control
     {
         public const string PART_DiagramControl = "PART_DiagramControl";
         public const string PART_ContentControl = "PART_ContentControl";