diff --git a/build.gradle b/build.gradle
index a12e12aa3bd..0b6c36fca1e 100644
--- a/build.gradle
+++ b/build.gradle
@@ -538,7 +538,10 @@ task collectBundleIntoDir(type: Copy) {
     from("scripts/sv", { into("scripts/sv") })
     from("scripts/cnv_wdl/", { into("scripts/cnv_wdl") })
     from("scripts/mutect2_wdl/", { into("scripts/mutect2_wdl") })
-    from("scripts/variantstore/", { into("scripts/variantstore") })
+    from("scripts/variantstore/") {
+        include 'wdl/extract/*.py'
+        exclude '**/test*.py'
+    } into("scripts/variantstore")
     from("scripts/dataproc-cluster-ui", { into("scripts/")})
     into "$buildDir/bundle-files-collected"
 }