diff --git a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/diagnostics/DiagnosticsCollector.kt b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/diagnostics/DiagnosticsCollector.kt
index 7b3d7688c0e..2d2b2b459ad 100644
--- a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/diagnostics/DiagnosticsCollector.kt
+++ b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/diagnostics/DiagnosticsCollector.kt
@@ -16,14 +16,15 @@ internal class DiagnosticsCollector(
     private val fileStructureCache: FileStructureCache,
     private val cache: ModuleFileCache,
 ) {
-    fun getDiagnosticsFor(element: KtElement, filter: DiagnosticCheckerFilter): List<FirPsiDiagnostic<*>> =
-        fileStructureCache
-            .getFileStructure(element.containingKtFile, cache)
-            .getStructureElementFor(element)
-            .diagnostics.diagnosticsFor(filter, element)
+    fun getDiagnosticsFor(element: KtElement, filter: DiagnosticCheckerFilter): List<FirPsiDiagnostic<*>> {
+        val fileStructure = fileStructureCache.getFileStructure(element.containingKtFile, cache)
+        val structureElement = fileStructure.getStructureElementFor(element)
+        val diagnostics = structureElement.diagnostics
+        return diagnostics.diagnosticsFor(filter, element)
+    }
 
-    fun collectDiagnosticsForFile(ktFile: KtFile, filter: DiagnosticCheckerFilter): Collection<FirPsiDiagnostic<*>> =
-        fileStructureCache
-            .getFileStructure(ktFile, cache)
-            .getAllDiagnosticsForFile(filter)
+    fun collectDiagnosticsForFile(ktFile: KtFile, filter: DiagnosticCheckerFilter): Collection<FirPsiDiagnostic<*>> {
+        val fileStructure = fileStructureCache.getFileStructure(ktFile, cache)
+        return fileStructure.getAllDiagnosticsForFile(filter)
+    }
 }
\ No newline at end of file