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;
         });