[modify] unit test fixed