fix a bug in last bug fix