diff --git a/CodeGenerator/Program.cs b/CodeGenerator/Program.cs index d1c69ca..3ef1a33 100644 --- a/CodeGenerator/Program.cs +++ b/CodeGenerator/Program.cs @@ -216,8 +216,10 @@ namespace CodeGenerator { } else if (listFields.Select(f => f.Item2).Contains(field)) { //update the size of the list attribute string fName = field.Name.ParseFName(); + //retrieve the name of the list pointer + string listPointerName = listFields.First(f => f.Item2.Name.ParseFName() == fName).Item1.Name.ParseFName(); //Try seeing if the list attribute is already present - var x = ctd.Members.OfType().FirstOrDefault(member => member.Name == fName); + var x = ctd.Members.OfType().FirstOrDefault(member => member.Name.ParseFName() == listPointerName); if (x != null) //Update the existing list attribute x.CustomAttributes[0].Arguments[5] = new(new CodePrimitiveExpression(totalSize));