Qu: About Morality If Punishing Criminals With severe or profound NPD And ASPD

Is it moral to punish and put people with Narcissistic Personality Disorder or Antisocial Personality Disorder with psychopathic features in jails or prisons instead of forensic correctional facilities?

First of all, jails or prisons don't have excellent training to handle those two personality disorders correctly if it is severe or profound, even if a prison psychiatrist or psychologist is trying to help severely mentally ill prisoners fix their thoughts and conduct.

Another issue is that people with severe or profound Narcissistic Personality Disorder or severe or profound Antisocial Personality Disorder with psychopathic features might be legally sane and competent to stand trial. Still, they have this strange nihilistic delusional view of this world. When they commit crimes, many also have a delusion that rules don't apply to them, due to disregard for right and wrong, and also black and withe thinking, with no grounds for the middle ground.

I hope you guys understand where I am coming from with this issue.