update unit test according to last bug fix