Fixed the small bug

Change-Id: If4828e96bf5f1f58baa96eadf279a4bafc8fa7a0