ASTNode (@Nullable ASTNode element) { if (element == null) { return null; } ASTNode parent = element.getTreeParent(); if (parent == null) { return null; } ASTNode grandParent = parent.getTreeParent(); if (is(grandParent, CLASS_SET)) { return grandParent; } return null; }