modify for fix build error

Change-Id: I896b00e56e139fe5c5726ebc390fdb6d8466a52a