fix the code

Change-Id: Ie40692315039c738c01cc18500208993bc5fece7