add potential bug fix