fix the code


Change-Id: I2a7437efc5fa6ccd768d134378e10450317d0044