modify for fix build error

Change-Id: I4806c46cd953f4d64325913d95faacd7298ec39b