Fix for making sure we are able to get the treatment factor values saved by the user

Reviewer: Tiff Go (post commit)

Issue: GCP-9787