Update the unit test according to new changes