diff --git a/build.mill b/build.mill
index 2fbcc301..ba1fbe19 100644
--- a/build.mill
+++ b/build.mill
@@ -122,7 +122,7 @@ trait SparkModule extends Cross.Module2[String, String] with SbtModule with Sona
     }
   }
 
-  object test extends SbtModuleTests with TestModule.ScalaTest {
+  object test extends SbtTests with TestModule.ScalaTest {
 
     override def millSourcePath = super.millSourcePath