some bug fixes

Change-Id: I834e94a505fdb209dc29793931a9c68b0e676111