Fixed hmm sampling for our case