private static void restoreComments(PsiForeachStatement foreachStatement, PsiStatement body) {
   final PsiElement parent = foreachStatement.getParent();
   for (PsiElement comment : PsiTreeUtil.findChildrenOfType(body, PsiComment.class)) {
     parent.addBefore(comment, foreachStatement);
   }
 }