diff --git a/uncompyle6/semantics/gencomp.py b/uncompyle6/semantics/gencomp.py index d1c63d4b..e144cd49 100644 --- a/uncompyle6/semantics/gencomp.py +++ b/uncompyle6/semantics/gencomp.py @@ -463,8 +463,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)