Fix exception handling

Change-Id: I13c21319031897efbeacc508821452e4663c033e