How to change AST Store context to Load when reusing a comprehension target?