diff --git a/GFramework.SourceGenerators.Tests/GFramework.SourceGenerators.Tests.csproj b/GFramework.SourceGenerators.Tests/GFramework.SourceGenerators.Tests.csproj
index de90d21..ec7d901 100644
--- a/GFramework.SourceGenerators.Tests/GFramework.SourceGenerators.Tests.csproj
+++ b/GFramework.SourceGenerators.Tests/GFramework.SourceGenerators.Tests.csproj
@@ -3,11 +3,13 @@
     <PropertyGroup>
         <ImplicitUsings>enable</ImplicitUsings>
         <Nullable>enable</Nullable>
-        <TargetFrameworks>net10.0;net8.0;net9.0</TargetFrameworks>
+        <TargetFramework>net8.0</TargetFramework>
     </PropertyGroup>
 
 
     <ItemGroup>
+        <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.14.0"/>
+        <PackageReference Include="Microsoft.CodeAnalysis.Common" Version="4.14.0"/>
         <PackageReference Include="Microsoft.CodeAnalysis.Common" Version="4.14.0"/>
         <PackageReference Include="Microsoft.CodeAnalysis.CSharp.SourceGenerators.Testing" Version="1.1.2"/>
         <PackageReference Include="Microsoft.NET.Test.Sdk" Version="18.0.1"/>
diff --git a/GFramework.SourceGenerators.Tests/rule/ContextAwareGeneratorTests.cs b/GFramework.SourceGenerators.Tests/rule/ContextAwareGeneratorTests.cs
index 3a70a92..c1b5b37 100644
--- a/GFramework.SourceGenerators.Tests/rule/ContextAwareGeneratorTests.cs
+++ b/GFramework.SourceGenerators.Tests/rule/ContextAwareGeneratorTests.cs
@@ -20,17 +20,27 @@ public class ContextAwareGeneratorTests
     {
         // 定义输入源代码，包含使用[ContextAware]特性的部分类
         const string source = """
-                              using GFramework.Core.rule;
-                              using GFramework.Core.architecture;
+                              using System;
+
+                              namespace GFramework.SourceGenerators.Attributes.rule
+                              {
+                                  [AttributeUsage(AttributeTargets.Class)]
+                                  public sealed class ContextAwareAttribute : Attribute
+                                  {
+                                  }
+                              }
 
                               namespace TestApp;
 
+                              using GFramework.SourceGenerators.Attributes.rule;
+
                               [ContextAware]
                               public partial class MyRule
                               {
                               }
                               """;
 
+
         // 定义期望的生成结果代码
         const string expected = """
                                 // <auto-generated/>
@@ -40,7 +50,6 @@ public class ContextAwareGeneratorTests
                                 partial class MyRule : GFramework.Core.rule.IContextAware
                                 {
                                     protected GFramework.Core.architecture.IArchitectureContext Context { get; private set; } = null!;
-
                                     void GFramework.Core.rule.IContextAware.SetContext(
                                         GFramework.Core.architecture.IArchitectureContext context)
                                     {
@@ -49,6 +58,7 @@ public class ContextAwareGeneratorTests
                                 }
                                 """;
 
+
         // 执行源代码生成器测试
         await GeneratorTest<ContextAwareGenerator>.RunAsync(
             source,
diff --git a/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs b/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
index 0fd1cb7..2c2de4f 100644
--- a/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
+++ b/GFramework.SourceGenerators/rule/ContextAwareGenerator.cs
@@ -99,13 +99,12 @@ public sealed class ContextAwareGenerator : IIncrementalGenerator
         sb.AppendLine(
             "    protected GFramework.Core.architecture.IArchitectureContext Context { get; private set; } = null!;");
 
-        sb.AppendLine("""
-                          void GFramework.Core.rule.IContextAware.SetContext(
-                              GFramework.Core.architecture.IArchitectureContext context)
-                          {
-                              Context = context;
-                          }
-                      """);
+        sb.AppendLine("    void GFramework.Core.rule.IContextAware.SetContext(");
+        sb.AppendLine("        GFramework.Core.architecture.IArchitectureContext context)");
+        sb.AppendLine("    {");
+        sb.AppendLine("        Context = context;");
+        sb.AppendLine("    }");
+
 
         sb.AppendLine("}");
 
diff --git a/GFramework.sln.DotSettings.user b/GFramework.sln.DotSettings.user
index 56537ed..8865a13 100644
--- a/GFramework.sln.DotSettings.user
+++ b/GFramework.sln.DotSettings.user
@@ -1,3 +1,8 @@
 <wpf:ResourceDictionary xml:space="preserve" xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" xmlns:s="clr-namespace:System;assembly=mscorlib" xmlns:ss="urn:shemas-jetbrains-com:settings-storage-xaml" xmlns:wpf="http://schemas.microsoft.com/winfx/2006/xaml/presentation">
 	<s:String x:Key="/Default/CodeInspection/ExcludedFiles/FilesAndFoldersToSkip2/=7020124F_002D9FFC_002D4AC3_002D8F3D_002DAAB8E0240759_002Ff_003ADiagnostic_002Ecs_002Fl_003AD_0021_003FTool_003FDevelopment_0020Tools_003FJetBrains_003F_002EJetBrains_003F_002ERider_003Fconfig_003Fresharper_002Dhost_003FDecompilerCache_003Fdecompiler_003F8fed4175e0a54d839582ab555761a2de4d5128_003Ff3_003Fd48d28bd_003FDiagnostic_002Ecs/@EntryIndexedValue">ForceIncluded</s:String>
-	<s:String x:Key="/Default/CodeInspection/ExcludedFiles/FilesAndFoldersToSkip2/=7020124F_002D9FFC_002D4AC3_002D8F3D_002DAAB8E0240759_002Ff_003AInputEventAction_002Ecs_002Fl_003AD_0021_003FTool_003FDevelopment_0020Tools_003FJetBrains_003F_002EJetBrains_003F_002ERider_003Fconfig_003Fresharper_002Dhost_003FDecompilerCache_003Fdecompiler_003F1c378f459c054fecaf4484a0fa6d44c055a800_003F18_003F33b52a1c_003FInputEventAction_002Ecs/@EntryIndexedValue">ForceIncluded</s:String></wpf:ResourceDictionary>
\ No newline at end of file
+	<s:String x:Key="/Default/CodeInspection/ExcludedFiles/FilesAndFoldersToSkip2/=7020124F_002D9FFC_002D4AC3_002D8F3D_002DAAB8E0240759_002Ff_003AInputEventAction_002Ecs_002Fl_003AD_0021_003FTool_003FDevelopment_0020Tools_003FJetBrains_003F_002EJetBrains_003F_002ERider_003Fconfig_003Fresharper_002Dhost_003FDecompilerCache_003Fdecompiler_003F1c378f459c054fecaf4484a0fa6d44c055a800_003F18_003F33b52a1c_003FInputEventAction_002Ecs/@EntryIndexedValue">ForceIncluded</s:String>
+	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=c33c8ad6_002Dd992_002D4817_002D855f_002D764f7c7897fa/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" IsActive="True" Name="Generates_ContextAware_Code" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;&#xD;
+  &lt;TestAncestor&gt;&#xD;
+    &lt;TestId&gt;NUnit3x::BB047F43-6AA0-4EA0-8AE9-E6B9784D9E8E::net8.0::GFramework.SourceGenerators.Tests.rule.ContextAwareGeneratorTests&lt;/TestId&gt;&#xD;
+  &lt;/TestAncestor&gt;&#xD;
+&lt;/SessionState&gt;</s:String></wpf:ResourceDictionary>
\ No newline at end of file