Fix the failure when doing unit test

Change-Id: Ib5df587535c1da8b2e324ae77c30718f20527599
CRs-Fixed: 652947