update test

Signed-off-by: Wang Zhen <yuchen12@gmail.com>