diff --git a/src/test/scala/TokenAggregatorTest.scala b/src/test/scala/TokenAggregatorTest.scala
index d37e8e9..aa4d7d2 100644
--- a/src/test/scala/TokenAggregatorTest.scala
+++ b/src/test/scala/TokenAggregatorTest.scala
@@ -91,7 +91,7 @@ class TokenAggregatorTest {
 
     var rowDataset = rowMemoryStream.toDF
 
-    val tokenAggregator = new TokenAggregator("_raw")
+    val tokenAggregator = new TokenAggregator("_raw", 32)
     val tokenAggregatorColumn = tokenAggregator.toColumn
 
     rowDataset = rowDataset