fix labels of treatment factors (after code review)
Reviewer: Dan Jao (post-commit)