Fix unit test issue

Change-Id: Ie0809591652a22379e59cd5f80b7cddb3896459d
2 files changed