diff --git a/kyuubi-common/src/test/scala/org/apache/kyuubi/util/ThreadUtilsSuite.scala b/kyuubi-common/src/test/scala/org/apache/kyuubi/util/ThreadUtilsSuite.scala
index 794c22b78..858a71570 100644
--- a/kyuubi-common/src/test/scala/org/apache/kyuubi/util/ThreadUtilsSuite.scala
+++ b/kyuubi-common/src/test/scala/org/apache/kyuubi/util/ThreadUtilsSuite.scala
@@ -25,10 +25,14 @@ class ThreadUtilsSuite extends KyuubiFunSuite {
 
   test("New daemon single thread scheduled executor") {
     val service = ThreadUtils.newDaemonSingleThreadScheduledExecutor("ThreadUtilsTest")
-    var flag = false
-    service.schedule(new Runnable {
-      override def run(): Unit = flag = true
-    }, 0, TimeUnit.MILLISECONDS)
-    assert(flag)
+    @volatile var threadName = ""
+    service.submit(new Runnable {
+      override def run(): Unit = {
+        threadName = Thread.currentThread().getName
+      }
+    })
+    service.shutdown()
+    service.awaitTermination(10, TimeUnit.SECONDS)
+    assert(threadName startsWith "ThreadUtilsTest")
   }
 }