DataContext (DataContext context) { PsiClass element = myUsageClass.getElement(); return element == null ? context : SimpleDataContext.getSimpleContext(LangDataKeys.TARGET_PSI_ELEMENT, element, context); }