private static void changeVisibility(JetChangeInfo changeInfo, PsiElement element) { JetModifierKeywordToken newVisibilityToken = JetRefactoringUtil.getVisibilityToken(changeInfo.getNewVisibility()); if (element instanceof JetFunction) { ((JetFunction) element).addModifier(newVisibilityToken); } else { JetPrimaryConstructor constructor = ((JetClass) element).getPrimaryConstructor(); assert constructor != null : "Primary constructor should be created before changing visibility"; constructor.addModifier(newVisibilityToken); } }