[YM] Updated the test