diff --git a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/inferred/InferredMetricsScraperTest.java b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/inferred/InferredMetricsScraperTest.java
index 6c4abb6be4..887a7555da 100644
--- a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/inferred/InferredMetricsScraperTest.java
+++ b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/inferred/InferredMetricsScraperTest.java
@@ -1,5 +1,7 @@
 package com.provectus.kafka.ui.service.metrics.scrape.inferred;
 
+import static com.provectus.kafka.ui.model.InternalLogDirStats.LogDirSpaceStats;
+import static com.provectus.kafka.ui.model.InternalLogDirStats.SegmentStats;
 import static com.provectus.kafka.ui.service.metrics.scrape.ScrapedClusterState.ConsumerGroupState;
 import static com.provectus.kafka.ui.service.metrics.scrape.ScrapedClusterState.NodeState;
 import static com.provectus.kafka.ui.service.metrics.scrape.ScrapedClusterState.TopicState;
@@ -29,8 +31,8 @@ class InferredMetricsScraperTest {
 
   @Test
   void allExpectedMetricsScraped() {
-    var segmentStats = new InternalLogDirStats.SegmentStats(1234L, 3);
-    var logDirStats = new InternalLogDirStats.LogDirSpaceStats(234L, 345L, Map.of(), Map.of());
+    var segmentStats = new SegmentStats(1234L, 3);
+    var logDirStats = new LogDirSpaceStats(234L, 345L, Map.of(), Map.of());
 
     Node node1 = new Node(1, "node1", 9092);
     Node node2 = new Node(2, "node2", 9092);
diff --git a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/prometheus/PrometheusMetricsRetrieverTest.java b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/prometheus/PrometheusMetricsRetrieverTest.java
index 4b5f96530a..7533385d20 100644
--- a/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/prometheus/PrometheusMetricsRetrieverTest.java
+++ b/kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/metrics/scrape/prometheus/PrometheusMetricsRetrieverTest.java
@@ -13,7 +13,6 @@ import okhttp3.mockwebserver.MockWebServer;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
-import org.springframework.web.reactive.function.client.WebClient;
 import reactor.test.StepVerifier;
 
 class PrometheusMetricsRetrieverTest {