Simplify logic for NA length