fix the code

Change-Id: Ia646b5d5bd31b09acf272e1a01212e36cb5ff71e