Fix mistake with new error catching