Added fixed for allowing to choose categorical variate in treatment factor

Reviewer: Tiff Go (post commit)

Issue: GCP-9768