fix the exception issue