fix bugs

Change-Id: Ie206e8a626b63eda64f177dd5e0528fb28c9c8b6