Add support for exit with error on connection drop