modify for fix build error

Change-Id: I3e2b36fe9742cb994c4937f83d8f75829741fffb