More fixes

Change-Id: Idd1388bf414177555144b788742a27b44ef7443f