HALO: fix for hidden state