diff --git a/distarray/dist/context.py b/distarray/dist/context.py index 1cccbaeb..1f0fe21b 100644 --- a/distarray/dist/context.py +++ b/distarray/dist/context.py @@ -622,7 +622,7 @@ def is_LocalArray(pxy): result = None else: if has_exactly_one(results): - result = next(x for x in result if x is not None) + result = next(x for x in results if x is not None) else: result = results