Fixing data labelling bug