Minor changes to test

Signed-off-by: Herbert <herberthb12@gmail.com>