fix labeling issue