Kiai bug fix