Fix UT with wrong message