fix the code

Change-Id: Ibe02a7f97581af4d5c361680c529d12d46d5a47d