Objectively there is no problem with that.
The problem is the "good and evil" view on mental health.
Separating sections inside the mental health hospital would be a good choice.
Would you really want outlaws to be considered "evil" because they had to go to the mental health hospital because of a crime, which was probably caused by a mental illness? Probably not.