fix the code

Change-Id: I9272caea5d85943a969e31de5b43d0d6ae4de983