fix normalize function to actually use the reverse_label_map