diff --git a/java/openmldb-batch/src/test/scala/com/_4paradigm/openmldb/batch/end2end/TestWindowUnion.scala b/java/openmldb-batch/src/test/scala/com/_4paradigm/openmldb/batch/end2end/TestWindowUnion.scala
index 7867e1a269e6df2a10c33471ce25075483146ec5..040c9bb3aa8236cfbd23ea42675241fbbda138f3 100644
--- a/java/openmldb-batch/src/test/scala/com/_4paradigm/openmldb/batch/end2end/TestWindowUnion.scala
+++ b/java/openmldb-batch/src/test/scala/com/_4paradigm/openmldb/batch/end2end/TestWindowUnion.scala
@@ -61,8 +61,7 @@ class TestWindowUnion extends SparkTestSuite {
     val outputDf = sess.sql(sqlText)
     val count = outputDf.count()
     val expectedCount = data.size
-    // TODO: Fix to mvn test in command-line
-    // assert(count == expectedCount)
+    assert(count == expectedCount)
   }
 
 }