PsiStatement (@Nullable PsiStatement statement) { if (statement instanceof PsiBlockStatement block) { PsiStatement onlyStatement = getOnlyStatementInBlock(block.getCodeBlock()); return (onlyStatement != null) ? onlyStatement : block; } else { return statement; } }