diff --git a/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java b/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java
index fdb7a6d833a..642bad98183 100644
--- a/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java
+++ b/integration/src/main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java
@@ -55,7 +55,7 @@ public void testPick() {
 
     @Test
     public void testGather() {
-        Engine engine = Engine.getEngine("PyTorch");
+        Engine engine = Engine.getEngine("MXNet");
         try (NDManager manager = engine.newBaseManager()) {
             NDArray arr = manager.arange(20f).reshape(-1, 4);
             long[] idx = {0, 0, 2, 1, 1, 2};