diff --git a/ExampleGenerator/Unity/Ui/UIBackingClassGenerator.cs b/ExampleGenerator/Unity/Ui/UIBackingClassGenerator.cs
index c8f0db2..dfadd9c 100644
--- a/ExampleGenerator/Unity/Ui/UIBackingClassGenerator.cs
+++ b/ExampleGenerator/Unity/Ui/UIBackingClassGenerator.cs
@@ -108,13 +108,8 @@ internal class {Helpers.UiElementAttribute} : Attribute
                 return null;
 
             List<ISymbol> elementFields = fieldsList.Where( f => GetUiElementAttributeData( f , uiElementAttributeSymbol ) != null ).ToList();
-
-            foreach ( var VARIABLE in elementFields )
-            {
-                //
-            }
-
-            var uxmlTraitFields = fieldsList.Where( f => GetUxmlTraitAttributeData( f , uxmlTraitAttributeSymbol ) != null ).ToList();
+            List<ISymbol> uxmlTraitFields = fieldsList.Where( f => GetUxmlTraitAttributeData( f , uxmlTraitAttributeSymbol ) != null ).ToList();
+            
             var source = new StringBuilder( $@"// <auto-generated/>
 
 using UnityEngine.UIElements;
@@ -146,7 +141,7 @@ public partial class {classSymbol.Name}
 
                 foreach ( ISymbol fieldSymbol in uxmlTraitFields )
                 {
-                    source.AppendLine( GetAttributeInitialization( fieldSymbol , uxmlTraitAttributeSymbol ) );
+                    source.AppendLine( GetAttributeInitialization( fieldSymbol ) );
                 }
 
                 source.Append( $@"        }}
@@ -333,7 +328,7 @@ public partial class {classSymbol.Name}
 
         private static string GetQualifyingTypeName( ITypeSymbol type ) { return type.ToDisplayString( SymbolDisplayFormat.FullyQualifiedFormat ); }
 
-        private string GetAttributeInitialization( ISymbol symbol , ISymbol attributeSymbol )
+        private string GetAttributeInitialization( ISymbol symbol )
         {
             // self.PlayerHealth = m_PlayerHealth.GetValueFromBag( bag , cc );
             return $"            self.{symbol.Name} = {GetAttributeDescriptionName( symbol.Name )}.GetValueFromBag( bag , cc );";