Updating error handling code.