Updat of the unit test