diff --git a/builder/map_attributes_to_ids.py b/builder/map_attributes_to_ids.py
index 06f8edd..80c1c16 100644
--- a/builder/map_attributes_to_ids.py
+++ b/builder/map_attributes_to_ids.py
@@ -9,7 +9,7 @@
 
 def main(filename, desc, mapping, output):
 
-    desc = pd.read_csv(desc, sep='\t', header=False, na_filter=False, names = ['ATTRIBUTE_ID', 'ATTRIBUTE_SYMBOL', 'DESCRIPTION'])
+    desc = pd.read_csv(desc, sep='\t', header=False, na_filter=False, names = ['ATTRIBUTE_ID', 'ATTRIBUTE_SYMBOL', 'ATTRIBUTE_NAME', 'DESCRIPTION'])
     desc['ATTRIBUTE_SYMBOL'] = desc['ATTRIBUTE_SYMBOL'].str.upper()
 
     symbols = pd.read_csv(mapping, sep='\t', header=None, na_filter=False, names=['NODE_ID', 'GENE_SYMBOL', 'SOURCE'])
@@ -42,4 +42,4 @@ def main(filename, desc, mapping, output):
     parser.add_argument('output', help='output file')
 
     args = parser.parse_args()
-    main(args.filename, args.desc, args.mapping, args.output)
\ No newline at end of file
+    main(args.filename, args.desc, args.mapping, args.output)