diff --git a/src/DKNet.FW.sln.DotSettings.user b/src/DKNet.FW.sln.DotSettings.user
index 18ad5218..f443ceb1 100644
--- a/src/DKNet.FW.sln.DotSettings.user
+++ b/src/DKNet.FW.sln.DotSettings.user
@@ -588,14 +588,14 @@
 	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=0109f02b_002D3fcd_002D4b82_002Db667_002Dfa31ec54c795/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" Name="All tests from &amp;lt;AspNet&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
   &lt;Solution /&gt;
 &lt;/SessionState&gt;</s:String>
-	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=076f3db0_002D9c90_002D464c_002Da713_002Df7c36319c497/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" Name="All tests from &amp;lt;AspNet&amp;gt;\&amp;lt;AspCore.Idempotency.Tests&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
+	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=076f3db0_002D9c90_002D464c_002Da713_002Df7c36319c497/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" IsActive="True" Name="All tests from &amp;lt;AspNet&amp;gt;\&amp;lt;AspCore.Idempotency.Tests&amp;gt;" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
   &lt;Or&gt;
     &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src/AspNet/AspCore.Idempotency.Tests" Presentation="&amp;lt;AspNet&amp;gt;\&amp;lt;AspCore.Idempotency.Tests&amp;gt;" /&gt;
     &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src/AspNet/AspCore.Idempotency.MsSqlStore.Tests" Presentation="&amp;lt;AspNet&amp;gt;\&amp;lt;AspCore.Idempotency.MsSqlStore.Tests&amp;gt;" /&gt;
   &lt;/Or&gt;
 &lt;/SessionState&gt;</s:String>
 	
-	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=bd972a6a_002D73e3_002D4962_002D9e9b_002Dc5100a3f62e4/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" IsActive="True" Name="Session" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
+	<s:String x:Key="/Default/Environment/UnitTesting/UnitTestSessionStore/Sessions/=bd972a6a_002D73e3_002D4962_002D9e9b_002Dc5100a3f62e4/@EntryIndexedValue">&lt;SessionState ContinuousTestingMode="0" Name="Session" xmlns="urn:schemas-jetbrains-com:jetbrains-ut-session"&gt;
   &lt;Or&gt;
     &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src" Presentation="&amp;lt;AspNet&amp;gt;" /&gt;
     &lt;Project Location="/Users/steven/_CODE/DRUNK/DKNet/src" Presentation="&amp;lt;Core&amp;gt;" /&gt;
diff --git a/src/EfCore/DKNet.EfCore.DtoGenerator/DKNet.EfCore.DtoGenerator.csproj b/src/EfCore/DKNet.EfCore.DtoGenerator/DKNet.EfCore.DtoGenerator.csproj
index 63765d71..67af29e9 100644
--- a/src/EfCore/DKNet.EfCore.DtoGenerator/DKNet.EfCore.DtoGenerator.csproj
+++ b/src/EfCore/DKNet.EfCore.DtoGenerator/DKNet.EfCore.DtoGenerator.csproj
@@ -18,7 +18,9 @@
         <!-- Suppress warnings about netstandard2.0 -->
         <NoWarn>$(NoWarn);NU5128</NoWarn>
     </PropertyGroup>
-
+    <ItemGroup>
+        <InternalsVisibleTo Include="EfCore.DtoGenerator.Tests"/>
+    </ItemGroup>
     <ItemGroup>
         <PackageReference Include="Microsoft.CodeAnalysis.CSharp" PrivateAssets="all"/>
         <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" PrivateAssets="all"/>
@@ -26,10 +28,10 @@
 
     <ItemGroup>
         <!-- Pack the generator DLL as an analyzer -->
-        <!--        <None Include="$(OutputPath)\$(AssemblyName).dll" Pack="true" PackagePath="analyzers/dotnet/cs" Visible="false"/>-->
+        <None Include="$(OutputPath)\$(AssemblyName).dll" Pack="true" PackagePath="analyzers/dotnet/cs" Visible="false"/>
 
         <!-- Include the attribute file in consuming projects -->
-        <!--        <None Include="GenerateDtoAttribute.cs" Pack="true" PackagePath="contentFiles/cs/any;content" BuildAction="Compile"/>-->
+        <None Include="GenerateDtoAttribute.cs" Pack="true" PackagePath="contentFiles/cs/any;content" BuildAction="Compile"/>
 
         <!-- Standard package files -->
         <None Include="..\..\NugetLogo.png" Pack="true" PackagePath="\"/>
diff --git a/src/EfCore/DKNet.EfCore.DtoGenerator/GenerateDtoAttribute.cs b/src/EfCore/DKNet.EfCore.DtoGenerator/GenerateDtoAttribute.cs
index 57e1464f..8d8428d0 100644
--- a/src/EfCore/DKNet.EfCore.DtoGenerator/GenerateDtoAttribute.cs
+++ b/src/EfCore/DKNet.EfCore.DtoGenerator/GenerateDtoAttribute.cs
@@ -35,7 +35,7 @@ namespace DKNet.EfCore.DtoGenerator;
 /// are automatically excluded, including both single entity properties and collection properties.
 /// </remarks>
 [AttributeUsage(AttributeTargets.Class | AttributeTargets.Struct, AllowMultiple = false, Inherited = false)]
-public sealed class GenerateDtoAttribute : Attribute
+internal sealed class GenerateDtoAttribute : Attribute
 {
     /// <summary>
     /// Gets the entity type to generate a DTO for.
diff --git a/src/EfCore/EfCore.DtoGenerator.Tests/EfCore.DtoGenerator.Tests.csproj b/src/EfCore/EfCore.DtoGenerator.Tests/EfCore.DtoGenerator.Tests.csproj
index 48ac8d12..e7baf20c 100644
--- a/src/EfCore/EfCore.DtoGenerator.Tests/EfCore.DtoGenerator.Tests.csproj
+++ b/src/EfCore/EfCore.DtoGenerator.Tests/EfCore.DtoGenerator.Tests.csproj
@@ -66,7 +66,10 @@
         <ProjectReference Include="..\DKNet.EfCore.DtoEntities\DKNet.EfCore.DtoEntities.csproj"/>
         <ProjectReference Include="..\DKNet.EfCore.DtoGenerator\DKNet.EfCore.DtoGenerator.csproj"
                           OutputItemType="Analyzer"
-                          ReferenceOutputAssembly="true"/>
+                          ReferenceOutputAssembly="true">
+            <PrivateAssets>all</PrivateAssets>
+            <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+        </ProjectReference>
     </ItemGroup>
 
     <ItemGroup>