diff --git a/uncompyle6/semantics/gencomp.py b/uncompyle6/semantics/gencomp.py index 78d8677d..c1018c90 100644 --- a/uncompyle6/semantics/gencomp.py +++ b/uncompyle6/semantics/gencomp.py @@ -465,8 +465,13 @@ class ComprehensionMixin: self.write(": ") self.preorder(n[1]) else: - if self.version == (3, 0) and len(n) > 1: - body = n[1] + if self.version == (3, 0): + if isinstance(n, Token): + body = store + elif len(n) > 1: + body = n[1] + else: + body = n[0] else: body = n[0] self.preorder(body)