diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamTest.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamTest.scala
index f9496520f383666499c04e138dbed8a49942b001..af2b58116b2aab2015ae36535a4a8762acf44b0b 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamTest.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/streaming/StreamTest.scala
@@ -189,7 +189,7 @@ trait StreamTest extends QueryTest with SharedSQLContext with Timeouts {
     }
 
     def apply(message: String)(condition: StreamExecution => Unit): AssertOnQuery = {
-      new AssertOnQuery(s => { condition; true }, message)
+      new AssertOnQuery(s => { condition(s); true }, message)
     }
   }