make some change for test case

Signed-off-by: Levi Wang <wangli5665@gmail.com>