Exactly. In concept, a judge can order psychological treatment, which should come from trained medical professionals. In practice, the US carceral state doesn’t have a good track record of improving the lives of people who fall into its grasp.