modify test code

Signed-off-by: yuanhuan <yuanhuan@tcl.com>