PyExpression () { ASTNode n = getNode().findChildByType(PythonDialectsTokenSetProvider.getInstance().getExpressionTokens()); if (followsNodeOfType(n, PyTokenTypes.FOR_KEYWORD)) { return (PyExpression)n.getPsi(); // can't be null, 'if' would fail } else return null; }