minor update to the unit test