[BUG_FIX] bug fix in test code