updated test code

Signed-off-by: K <mustkre8@gmail.com>