Fix bugs

Change-Id: I4d628f62e3941efaf2cf680470ea0ce8ffcf48f0