Updates to tests

Signed-off-by: pris <priscilla.ngsc@gmail.com>