diff --git a/data/cropped_regions_cached_features.py b/data/cropped_regions_cached_features.py
index ce4dd3f4701a6d388f295822318a91ca9af7c110..2aa8617a4518797e44ca984303814c47feb274c4 100644
--- a/data/cropped_regions_cached_features.py
+++ b/data/cropped_regions_cached_features.py
@@ -212,7 +212,8 @@ class data():
         #     label_id = self.object_labels_dict[_unknown_token]
         #     object_label_encoding[0,label_id] = 1.0
 
-        return object_label_encoding/np.sum(object_label_encoding), object_labels
+        #return object_label_encoding/np.sum(object_label_encoding), object_labels
+        return object_label_encoding, object_labels
             
     def get_attribute_label(self, sample):
         # Attribute is turned on if it is present