diff --git a/dev/mvn-build-helper/assembly/pom.xml b/dev/mvn-build-helper/assembly/pom.xml
index 935059be..dfbd574e 100644
--- a/dev/mvn-build-helper/assembly/pom.xml
+++ b/dev/mvn-build-helper/assembly/pom.xml
@@ -13,6 +13,11 @@
   <packaging>jar</packaging>
 
   <dependencies>
+    <dependency>
+      <groupId>org.blaze</groupId>
+      <artifactId>spark-extension</artifactId>
+      <version>${revision}</version>
+    </dependency>
     <dependency>
       <groupId>org.blaze</groupId>
       <artifactId>${shimPkg}</artifactId>