diff --git a/src/SourceGeneratorTestHelpers.MSTest/GeneratorDriverRunResultExtensions.cs b/src/SourceGeneratorTestHelpers.MSTest/GeneratorDriverRunResultExtensions.cs
index 7102622..3596c47 100644
--- a/src/SourceGeneratorTestHelpers.MSTest/GeneratorDriverRunResultExtensions.cs
+++ b/src/SourceGeneratorTestHelpers.MSTest/GeneratorDriverRunResultExtensions.cs
@@ -20,6 +20,6 @@ public static void ShouldProduce(this GeneratorDriverRunResult result, string fi
         if (result is null)
             throw new ArgumentNullException(nameof(result));
 #endif
-        result.ShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new AssertFailedException(message));
+        result.InternalShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new AssertFailedException(message));
     }
-}
\ No newline at end of file
+}
diff --git a/src/SourceGeneratorTestHelpers.MSTest/SourceGeneratorTestHelpers.MSTest.csproj b/src/SourceGeneratorTestHelpers.MSTest/SourceGeneratorTestHelpers.MSTest.csproj
index 2a089f2..b8af3b3 100644
--- a/src/SourceGeneratorTestHelpers.MSTest/SourceGeneratorTestHelpers.MSTest.csproj
+++ b/src/SourceGeneratorTestHelpers.MSTest/SourceGeneratorTestHelpers.MSTest.csproj
@@ -6,7 +6,7 @@
         <Nullable>enable</Nullable>
         <RootNamespace>SourceGeneratorTestHelpers.MSTest</RootNamespace>
         <LangVersion>latest</LangVersion>
-        <Version>8.0.5</Version>
+        <Version>8.0.6</Version>
         <Title>SourceGeneratorTestHelpers.MSTest</Title>
         <Authors>Jean-Sebastien Carle</Authors>
         <Description>Test helpers and extension methods to simplify testing of .NET source generators.</Description>
@@ -17,9 +17,9 @@
         <PackageReadmeFile>README.md</PackageReadmeFile>
         <RepositoryUrl>https://github.com/jscarle/SourceGeneratorTestHelpers</RepositoryUrl>
         <RepositoryType>git</RepositoryType>
-        <PackageTags>testing source-generators</PackageTags>
-        <AssemblyVersion>8.0.5.0</AssemblyVersion>
-        <FileVersion>8.0.5.0</FileVersion>
+        <PackageTags>testing source-generators mstest</PackageTags>
+        <AssemblyVersion>8.0.6.0</AssemblyVersion>
+        <FileVersion>8.0.6.0</FileVersion>
         <NeutralLanguage>en-US</NeutralLanguage>
         <IncludeSymbols>true</IncludeSymbols>
         <SymbolPackageFormat>snupkg</SymbolPackageFormat>
@@ -32,7 +32,7 @@
 
     <ItemGroup>
         <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.3.1"/>
-        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.5"/>
+        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.6" />
         <PackageReference Include="MSTest.TestFramework" Version="2.0.0" />
     </ItemGroup>
 
diff --git a/src/SourceGeneratorTestHelpers.NUnit/GeneratorDriverRunResultExtensions.cs b/src/SourceGeneratorTestHelpers.NUnit/GeneratorDriverRunResultExtensions.cs
index ada6cd7..3c56989 100644
--- a/src/SourceGeneratorTestHelpers.NUnit/GeneratorDriverRunResultExtensions.cs
+++ b/src/SourceGeneratorTestHelpers.NUnit/GeneratorDriverRunResultExtensions.cs
@@ -20,6 +20,6 @@ public static void ShouldProduce(this GeneratorDriverRunResult result, string fi
         if (result is null)
             throw new ArgumentNullException(nameof(result));
 #endif
-        result.ShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new AssertionException(message));
+        result.InternalShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new AssertionException(message));
     }
-}
\ No newline at end of file
+}
diff --git a/src/SourceGeneratorTestHelpers.NUnit/SourceGeneratorTestHelpers.NUnit.csproj b/src/SourceGeneratorTestHelpers.NUnit/SourceGeneratorTestHelpers.NUnit.csproj
index 0bcb51b..892241e 100644
--- a/src/SourceGeneratorTestHelpers.NUnit/SourceGeneratorTestHelpers.NUnit.csproj
+++ b/src/SourceGeneratorTestHelpers.NUnit/SourceGeneratorTestHelpers.NUnit.csproj
@@ -6,7 +6,7 @@
         <Nullable>enable</Nullable>
         <RootNamespace>SourceGeneratorTestHelpers.NUnit</RootNamespace>
         <LangVersion>latest</LangVersion>
-        <Version>8.0.5</Version>
+        <Version>8.0.6</Version>
         <Title>SourceGeneratorTestHelpers.NUnit</Title>
         <Authors>Jean-Sebastien Carle</Authors>
         <Description>Test helpers and extension methods to simplify testing of .NET source generators.</Description>
@@ -17,9 +17,9 @@
         <PackageReadmeFile>README.md</PackageReadmeFile>
         <RepositoryUrl>https://github.com/jscarle/SourceGeneratorTestHelpers</RepositoryUrl>
         <RepositoryType>git</RepositoryType>
-        <PackageTags>testing source-generators</PackageTags>
-        <AssemblyVersion>8.0.5.0</AssemblyVersion>
-        <FileVersion>8.0.5.0</FileVersion>
+        <PackageTags>testing source-generators nunit</PackageTags>
+        <AssemblyVersion>8.0.6.0</AssemblyVersion>
+        <FileVersion>8.0.6.0</FileVersion>
         <NeutralLanguage>en-US</NeutralLanguage>
         <IncludeSymbols>true</IncludeSymbols>
         <SymbolPackageFormat>snupkg</SymbolPackageFormat>
@@ -32,7 +32,7 @@
 
     <ItemGroup>
         <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.3.1"/>
-        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.5"/>
+        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.6"/>
         <PackageReference Include="NUnit" Version="3.0.0"/>
     </ItemGroup>
 
diff --git a/src/SourceGeneratorTestHelpers.XUnit/GeneratorDriverRunResultExtensions.cs b/src/SourceGeneratorTestHelpers.XUnit/GeneratorDriverRunResultExtensions.cs
index c275890..aa27547 100644
--- a/src/SourceGeneratorTestHelpers.XUnit/GeneratorDriverRunResultExtensions.cs
+++ b/src/SourceGeneratorTestHelpers.XUnit/GeneratorDriverRunResultExtensions.cs
@@ -20,6 +20,6 @@ public static void ShouldProduce(this GeneratorDriverRunResult result, string fi
         if (result is null)
             throw new ArgumentNullException(nameof(result));
 #endif
-        result.ShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new XunitException(message));
+        result.InternalShouldProduce(filePathEndsWith, expectedSource, assertOnErrors, message => throw new XunitException(message));
     }
-}
\ No newline at end of file
+}
diff --git a/src/SourceGeneratorTestHelpers.XUnit/SourceGeneratorTestHelpers.XUnit.csproj b/src/SourceGeneratorTestHelpers.XUnit/SourceGeneratorTestHelpers.XUnit.csproj
index cc5102f..428cfa3 100644
--- a/src/SourceGeneratorTestHelpers.XUnit/SourceGeneratorTestHelpers.XUnit.csproj
+++ b/src/SourceGeneratorTestHelpers.XUnit/SourceGeneratorTestHelpers.XUnit.csproj
@@ -6,7 +6,7 @@
         <Nullable>enable</Nullable>
         <RootNamespace>SourceGeneratorTestHelpers.XUnit</RootNamespace>
         <LangVersion>latest</LangVersion>
-        <Version>8.0.5</Version>
+        <Version>8.0.6</Version>
         <Title>SourceGeneratorTestHelpers.XUnit</Title>
         <Authors>Jean-Sebastien Carle</Authors>
         <Description>Test helpers and extension methods to simplify testing of .NET source generators.</Description>
@@ -17,9 +17,9 @@
         <PackageReadmeFile>README.md</PackageReadmeFile>
         <RepositoryUrl>https://github.com/jscarle/SourceGeneratorTestHelpers</RepositoryUrl>
         <RepositoryType>git</RepositoryType>
-        <PackageTags>testing source-generators</PackageTags>
-        <AssemblyVersion>8.0.5.0</AssemblyVersion>
-        <FileVersion>8.0.5.0</FileVersion>
+        <PackageTags>testing source-generators xunit</PackageTags>
+        <AssemblyVersion>8.0.6.0</AssemblyVersion>
+        <FileVersion>8.0.6.0</FileVersion>
         <NeutralLanguage>en-US</NeutralLanguage>
         <IncludeSymbols>true</IncludeSymbols>
         <SymbolPackageFormat>snupkg</SymbolPackageFormat>
@@ -32,7 +32,7 @@
 
     <ItemGroup>
         <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.3.1"/>
-        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.5"/>
+        <PackageReference Include="SourceGeneratorTestHelpers" Version="8.0.6"/>
         <PackageReference Include="xunit.assert" Version="2.0.0"/>
     </ItemGroup>