diff --git a/src/test/scala/com/databricks/labs/mosaic/expressions/geometry/ST_ZBehaviors.scala b/src/test/scala/com/databricks/labs/mosaic/expressions/geometry/ST_ZBehaviors.scala
index 566b74b70..3ce7ad57d 100644
--- a/src/test/scala/com/databricks/labs/mosaic/expressions/geometry/ST_ZBehaviors.scala
+++ b/src/test/scala/com/databricks/labs/mosaic/expressions/geometry/ST_ZBehaviors.scala
@@ -31,7 +31,7 @@ trait ST_ZBehaviors extends MosaicSpatialQueryTest {
             .withColumn("result", st_z($"wkt"))
             .where($"expected" === $"result")
 
-        results.count shouldBe 4
+        result.count shouldBe 4
     }
 
     def stzCodegen(mosaicContext: MosaicContext): Unit = {