modify for fix build error

Change-Id: I8ffa7a1e2577338f69ff5a4e1e918e546f25e7cc