For proper setting of treatment factor label

GCP-9802

Code review : Daniel Jao