diff --git a/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java b/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
index 160b2fa4d..0f5cbefcf 100644
--- a/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
+++ b/src/main/java/org/opensearch/neuralsearch/common/MinClusterVersionUtil.java
@@ -6,7 +6,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import org.opensearch.Version;
-import org.opensearch.knn.index.IndexUtil;
+import org.opensearch.knn.index.util.IndexUtil;
 import org.opensearch.neuralsearch.util.NeuralSearchClusterUtil;
 
 import java.util.Map;
diff --git a/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java b/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
index 3220ce8c6..2e9183682 100644
--- a/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
+++ b/src/test/java/org/opensearch/neuralsearch/NeuralSearchTests.java
@@ -5,7 +5,7 @@
 package org.opensearch.neuralsearch;
 
 import org.opensearch.knn.common.KNNConstants;
-import org.opensearch.knn.index.util.KNNEngine;
+import org.opensearch.knn.index.engine.KNNEngine;
 import org.opensearch.test.OpenSearchTestCase;
 
 public class NeuralSearchTests extends OpenSearchTestCase {