Kriminalvården är därmed huvudansvarig myndighet för tilllämpningen av häkteslagen beträffande samtliga personer som förvaras i häkte eller i kriminalvårdsanstalt samt beträffande personer som är intagna i häkte men vistas på sjukhus , med undantag för dem som överförs till rättspsykiatriska undersökningsenheter eller till vissa rättspsykiatriska vårdenheter .