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);
    }
  }