[test] Updated the tests