From b1615a8233d919bce422b84d70502e6bd094f760 Mon Sep 17 00:00:00 2001
From: "henry.5" <henry.5@kakaoenterprise.com>
Date: Wed, 6 Aug 2025 23:59:56 +0900
Subject: [PATCH] ZEPPELIN-6238 Modified test code

---
 .../java/org/apache/zeppelin/rest/NotebookRestApiTest.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/zeppelin-server/src/test/java/org/apache/zeppelin/rest/NotebookRestApiTest.java b/zeppelin-server/src/test/java/org/apache/zeppelin/rest/NotebookRestApiTest.java
index 5a97b3db81d..2a4ad37d3e7 100644
--- a/zeppelin-server/src/test/java/org/apache/zeppelin/rest/NotebookRestApiTest.java
+++ b/zeppelin-server/src/test/java/org/apache/zeppelin/rest/NotebookRestApiTest.java
@@ -666,12 +666,12 @@ void testRunNoteNonBlocking() throws Exception {
           Paragraph p2 = note1.getParagraph(1);
           try {
             p1.waitUntilFinished();
+            assertEquals(Job.Status.FINISHED, p1.getStatus());
             p2.waitUntilFinished();
+            assertEquals(Job.Status.FINISHED, p2.getStatus());
           } catch (InterruptedException e) {
             fail(e);
           }
-          assertEquals(Job.Status.FINISHED, p1.getStatus());
-          assertEquals(Job.Status.FINISHED, p2.getStatus());
           assertEquals("hello\n", p2.getReturn().message().get(0).getData());
           return null;
         });