diff --git a/test/BlueWest.MapTo.Tests/Infrastructure/CSharpGenerator.cs b/test/BlueWest.MapTo.Tests/Infrastructure/CSharpGenerator.cs
index af9eca0..b43e274 100644
--- a/test/BlueWest.MapTo.Tests/Infrastructure/CSharpGenerator.cs
+++ b/test/BlueWest.MapTo.Tests/Infrastructure/CSharpGenerator.cs
@@ -50,7 +50,7 @@ namespace MapTo.Tests.Infrastructure
             }
 
             var driver = CSharpGeneratorDriver.Create(
-                new List<ISourceGenerator>() { new MapToGenerator(), new EfAddMethodsGenerator() },
+                new List<ISourceGenerator>() { new MapToGenerator(), new EfMethodsGenerator() },
                 optionsProvider: new TestAnalyzerConfigOptionsProvider(analyzerConfigOptions),
                 parseOptions: new CSharpParseOptions(languageVersion)
             );