Report correct stats

Author: inputfalken

src/DynamoDBGenerator.SourceGenerator/DynamoDBDMarshaller.cs

@@ -109,7 +109,7 @@ static IEnumerable<DynamoDBMarshallerArguments> Arguments(
                     return new DynamoDBMarshallerArguments(
                         entityTypeSymbol,
                         x.NamedArguments
-                            .Where(x => x.Key is ArgumentType)
+                            .Where(y => y.Key is ArgumentType)
                             .Cast<KeyValuePair<string, TypedConstant>?>()
                             .FirstOrDefault() is { } argumentType
                             ? argumentType.Value is { Value: INamedTypeSymbol namedTypeSymbol }

src/DynamoDBGenerator.SourceGenerator/MarshallerFactory.cs

@@ -85,7 +85,7 @@ MarshallerOptions options
 using {Constants.DynamoDBGenerator.Namespace.ExceptionsFullName};
 using {Constants.DynamoDBGenerator.Namespace.InternalFullName};";
 
-        return originatingType
+        var marshaller = originatingType
             .NamespaceDeclaration(
                 originatingType
                     .TypeDeclaration()
@@ -99,9 +99,12 @@ MarshallerOptions options
                     .Concat(Marshaller.CreateClass(arguments, dynamoDbProperties, options))
                     .Concat(UnMarshaller.CreateClass(arguments, dynamoDbProperties, options))
             )
-            .Prepend(res)
-            .Prepend(
-                $"// <auto-generated | Duration {TimeSpan.FromTicks(Stopwatch.GetTimestamp() - timestamp)} | TimeStamp: {DateTime.Now:yyyy-MM-dd HH:mm:ss}>"
-            );
+            .Prepend(res);
+        
+        
+        foreach (var x in marshaller)
+            yield return x;
+        yield return
+            $"// <auto-generated | Duration {TimeSpan.FromTicks(Stopwatch.GetTimestamp() - timestamp)} | TimeStamp: {DateTime.Now:yyyy-MM-dd HH:mm:ss}>";
     }
 }