Fix mislabeling issue