diff --git a/json/src/com/jetbrains/jsonSchema/fus/JsonSchemaHighlightingSessionStatisticsCollector.kt b/json/src/com/jetbrains/jsonSchema/fus/JsonSchemaHighlightingSessionStatisticsCollector.kt
index 02229010aca9..9ed599835b8e 100644
--- a/json/src/com/jetbrains/jsonSchema/fus/JsonSchemaHighlightingSessionStatisticsCollector.kt
+++ b/json/src/com/jetbrains/jsonSchema/fus/JsonSchemaHighlightingSessionStatisticsCollector.kt
@@ -39,7 +39,7 @@ internal class JsonSchemaHighlightingSessionStatisticsCollector {
     }
   }
 
-  fun reportSchemaUsageFeature(featureKind: JsonSchemaFusFeature) {
+  fun reportSchemaUsageFeature(featureKind: JsonSchemaFusCountedFeature) {
     val currentSession = getCurrentSession() ?: return
     currentSession.featuresWithCount[featureKind] = currentSession.featuresWithCount.getOrDefault(featureKind, 0) + 1
   }
diff --git a/json/tests/test/com/jetbrains/jsonSchema/JsonSchemaPerformanceTest.java b/json/tests/test/com/jetbrains/jsonSchema/JsonSchemaPerformanceTest.java
index bd70788dca90..ee2eda87c552 100644
--- a/json/tests/test/com/jetbrains/jsonSchema/JsonSchemaPerformanceTest.java
+++ b/json/tests/test/com/jetbrains/jsonSchema/JsonSchemaPerformanceTest.java
@@ -52,7 +52,7 @@ public class JsonSchemaPerformanceTest extends JsonSchemaHeavyAbstractTest {
       registerJsonSchema(myFixture, schemaText, "json", it -> true);
       myFixture.configureByFile("/azure-file.json");
       myFixture.checkHighlighting(true, false, true);
-    }).start();
+    }).attempts(5).start();
   }
 
   public void testSwaggerHighlighting() {