Fix error handle bug