From 63c25af5baf6162f062601d42afc29eede0d8f44 Mon Sep 17 00:00:00 2001
From: KexinFeng <fengx463@umn.edu>
Date: Wed, 29 Jun 2022 10:22:11 -0700
Subject: [PATCH] test fix

---
 .../main/java/ai/djl/integration/tests/ndarray/NDIndexTest.java  | 1 +
 1 file changed, 1 insertion(+)

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 863873a7623f..c9c6744bd818 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
@@ -159,6 +159,7 @@ public void testSetArray() {
             Assert.assertEquals(original, expected);
 
             // set by index array
+            original = manager.arange(1, 10).reshape(3, 3);
             NDArray index = manager.create(new long[] {0, 1}, new Shape(2));
             value = manager.create(new int[] {666, 777, 888, 999}, new Shape(2, 2));
             original.set(new NDIndex("{}, :{}", index, 2), value);