diff --git a/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/CmsKitContentWidgetOptions.cs b/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/CmsKitContentWidgetOptions.cs
index 2d2ed68302..1f9dae60ab 100644
--- a/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/CmsKitContentWidgetOptions.cs
+++ b/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/CmsKitContentWidgetOptions.cs
@@ -11,8 +11,8 @@ public class CmsKitContentWidgetOptions
         WidgetConfigs = new();
     }
 
-    public void AddWidgetConfig(string name, ContentWidgetConfig contentWidgetConfig)
+    public void AddWidget(string widgetName, string widgetKey)
     {
-        WidgetConfigs.Add(name, contentWidgetConfig);
+        WidgetConfigs.Add(widgetName, new ContentWidgetConfig(widgetKey));
     }
 }
\ No newline at end of file
diff --git a/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/ContentWidgetConfig.cs b/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/ContentWidgetConfig.cs
index 8dc1f5d29f..a5eab5930a 100644
--- a/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/ContentWidgetConfig.cs
+++ b/modules/cms-kit/src/Volo.CmsKit.Common.Application/Volo/CmsKit/Contents/ContentWidgetConfig.cs
@@ -1,14 +1,11 @@
-using System;
-
-namespace Volo.CmsKit.Contents;
+namespace Volo.CmsKit.Contents;
 
 public class ContentWidgetConfig
 {
     public string Name { get; }
-    public Type ViewComponentType { get; set; } //TODO: Remove this
 
-    public ContentWidgetConfig(string name) //TODO: widgetName
+    public ContentWidgetConfig(string widgetName)
     {
-        Name = name;
+        Name = widgetName;
     }
 }
\ No newline at end of file
diff --git a/modules/cms-kit/test/Volo.CmsKit.Application.Tests/Contents/ContentParser_Test.cs b/modules/cms-kit/test/Volo.CmsKit.Application.Tests/Contents/ContentParser_Test.cs
index 7deeff2a80..108d883a70 100644
--- a/modules/cms-kit/test/Volo.CmsKit.Application.Tests/Contents/ContentParser_Test.cs
+++ b/modules/cms-kit/test/Volo.CmsKit.Application.Tests/Contents/ContentParser_Test.cs
@@ -20,8 +20,8 @@ public class ContentParser_Test : CmsKitDomainTestBase
     [Fact]
     public async Task ParseAsync_ShouldWorkWithDifferentWidgetTypes()
     {
-        _options.Value.AddWidgetConfig(testData.PollName, new ContentWidgetConfig(testData.WidgetName));
-        _options.Value.AddWidgetConfig("ImageGallery", new ContentWidgetConfig("ImageGallery"));//test
+        _options.Value.AddWidget(testData.PollName, testData.WidgetName);
+        _options.Value.AddWidget("ImageGallery", "ImageGallery");
         contentParser = new ContentParser(_options);
 
         var content = @"**ABP Framework** is completely open source and developed in a community-driven manner.
@@ -54,7 +54,7 @@ public class ContentParser_Test : CmsKitDomainTestBase
     [Fact]
     public async Task ParseAsync_ShouldWorkWithWrongConfigOptions()
     {
-        _options.Value.AddWidgetConfig(testData.WidgetName, new ContentWidgetConfig(testData.PollName));
+        _options.Value.AddWidget(testData.WidgetName, testData.PollName);
         contentParser = new ContentParser(_options);
 
         var content = @"**ABP Framework** is completely open source and developed in a community-driven manner.
@@ -70,7 +70,7 @@ public class ContentParser_Test : CmsKitDomainTestBase
     [Fact]
     public async Task ParseAsync_ShouldWorkWithWrongWidgetType()
     {
-        _options.Value.AddWidgetConfig(testData.PollName, new ContentWidgetConfig(testData.WidgetName));
+        _options.Value.AddWidget(testData.PollName, testData.WidgetName);
         contentParser = new ContentParser(_options);
 
         var content = @"**ABP Framework** is completely open source and developed in a community-driven manner.
@@ -86,7 +86,7 @@ public class ContentParser_Test : CmsKitDomainTestBase
     [Fact]
     public async Task ParseAsync_ShouldWorkWithWrongPollName()
     {
-        _options.Value.AddWidgetConfig(testData.PollName, new ContentWidgetConfig(testData.WidgetName));
+        _options.Value.AddWidget(testData.PollName, testData.WidgetName);
         contentParser = new ContentParser(_options);
 
         var content = @"**ABP Framework** is completely open source and developed in a community-driven manner.
@@ -103,7 +103,7 @@ public class ContentParser_Test : CmsKitDomainTestBase
     [MemberData(nameof(ExampleData))]
     public async Task ParseAsync_ShouldWorkProperlyWithCorrectInputs(string content, int expectedLine)
     {
-        _options.Value.AddWidgetConfig(testData.PollName, new ContentWidgetConfig(testData.WidgetName));
+        _options.Value.AddWidget(testData.PollName, testData.WidgetName);
         contentParser = new ContentParser(_options);
 
         var widgets = await contentParser.ParseAsync(content);