fix UT compilation error

Change-Id: Ie0c2b8c7f48a86a3f8d35b7b526637821e0ed87b