diff --git a/dsls/sql/src/test/java/org/apache/beam/dsls/sql/BeamSqlDslUdfUdafTest.java b/dsls/sql/src/test/java/org/apache/beam/dsls/sql/BeamSqlDslUdfUdafTest.java
index e517065a48ddd..ddc0c9c02c8e8 100644
--- a/dsls/sql/src/test/java/org/apache/beam/dsls/sql/BeamSqlDslUdfUdafTest.java
+++ b/dsls/sql/src/test/java/org/apache/beam/dsls/sql/BeamSqlDslUdfUdafTest.java
@@ -87,6 +87,8 @@ public void testUdf() throws Exception{
         .apply("testUdf2",
             BeamSql.query(sql2).withUdf("cubic2", CubicInteger.class, "cubic"));
     PAssert.that(result2).containsInAnyOrder(record);
+
+    pipeline.run().waitUntilFinish();
   }
 
   /**