modify for fix build error

Change-Id: I931b3504f812c5aa11ff3cdc876e2b1e88aa955d