diff --git a/adversarial_summary.py b/adversarial_summary.py
index b52cd8b..eb63d89 100644
--- a/adversarial_summary.py
+++ b/adversarial_summary.py
@@ -19,13 +19,13 @@ for file in ['train', 'val_unseen', 'val_seen', 'train_seen', 'test', 'val_train
                     'path': i['path'],
                     'heading': float(i['heading']),
                     'instructions': [ i['instruction'] ],
-                    'swap': [ True if i['found'] else False ],
+                    'found': [ i['found'] ],
                     'id': i['id'],
                     'objId': i['objId']
                 }
             else:
                 result[instruction_id]['instructions'].append(i['instruction'])
-                result[instruction_id]['swap'].append( True if i['found'] else False)
+                result[instruction_id]['found'].append( i['found'] )
 
         output = []
         for k, item in result.items():