diff --git a/src/FluentCommand.Generators/DataReaderFactoryWriter.cs b/src/FluentCommand.Generators/DataReaderFactoryWriter.cs
index b429b537..75ee22c3 100644
--- a/src/FluentCommand.Generators/DataReaderFactoryWriter.cs
+++ b/src/FluentCommand.Generators/DataReaderFactoryWriter.cs
@@ -8,7 +8,7 @@ public static class DataReaderFactoryWriter
     private static int _counter = 0;
 #endif
 
-    public static string Generate(EntityClass entityClass, bool skipVersion = false)
+    public static string Generate(EntityClass entityClass)
     {
         if (entityClass == null)
             throw new ArgumentNullException(nameof(entityClass));
@@ -38,12 +38,12 @@ public static string Generate(EntityClass entityClass, bool skipVersion = false)
             .AppendLine("/// Extension methods for FluentCommand")
             .AppendLine("/// </summary>");
 
-        //codeBuilder
-        //    .Append("[global::System.CodeDom.Compiler.GeneratedCode(\"")
-        //    .Append(ThisAssembly.Project.AssemblyName)
-        //    .Append("\", \"")
-        //    .Append(skipVersion ? "1.0.0.0" : ThisAssembly.Info.Version)
-        //    .AppendLine("\")]");
+        codeBuilder
+            .Append("[global::System.CodeDom.Compiler.GeneratedCode(\"")
+            .Append("FluentCommand.Generators")
+            .Append("\", \"")
+            .Append("1.0.0.0")
+            .AppendLine("\")]");
 
         codeBuilder
             .AppendLine("[global::System.Diagnostics.DebuggerNonUserCodeAttribute]")
diff --git a/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs b/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs
index 7f2b6bc6..f0682ba3 100644
--- a/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs
+++ b/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs
@@ -24,7 +24,7 @@ public async Task Generate()
             }.ToImmutableArray()
         );
 
-        var source = DataReaderFactoryWriter.Generate(entityClass, true);
+        var source = DataReaderFactoryWriter.Generate(entityClass);
 
         await Verifier
             .Verify(source)