diff --git a/AIStudio.Wpf.BaseDiagram/Helpers/EnumHelper.cs b/AIStudio.Wpf.BaseDiagram/Helpers/EnumHelper.cs
index 4cf2eca..09591f5 100644
--- a/AIStudio.Wpf.BaseDiagram/Helpers/EnumHelper.cs
+++ b/AIStudio.Wpf.BaseDiagram/Helpers/EnumHelper.cs
@@ -16,7 +16,7 @@ namespace AIStudio.Wpf.BaseDiagram.Helpers
             return (Type)obj.GetValue(EnumProperty);
         }
 
-        public static void SetEnum(DependencyObject obj, string value)
+        public static void SetEnum(DependencyObject obj, Type value)
         {
             obj.SetValue(EnumProperty, value);
         }