[YM] update the tests to verify both the test passes