fix pos_hidden_states's +1 error and add print