BUG: Fixed dimension error in computing reject