PyElement (@NotNull PyReferenceExpression node) { final PyElement resolve = PyUtil.as(node.getReference(getResolveContext()).resolve(), PyElement.class); return resolve == null ? null : PyiUtil.getOriginalElementOrLeaveAsIs(resolve, PyElement.class); }