fix the unittest failed

Change-Id: I4a56de0c67bfb6dac35b020b1f0cdfc452c3dd2c