Fix labelling issue