fixes labelling bug