Fix log_trans_prob input bug