modify for fix build error

Change-Id: Ib6555c971482f149b916b2e76c35120718f80e1b