diff --git a/build.mill b/build.mill
index 6c97279..15551fd 100644
--- a/build.mill
+++ b/build.mill
@@ -2,7 +2,7 @@
 
 import mill._, javalib._, errorprone._
 
-object `package` extends RootModule with JavaModule with ErrorProneModule {
+object `package` extends RootModule with MavenModule with ErrorProneModule {
 
   def artifactName = "vavr-effect"
 
@@ -25,28 +25,19 @@
     Agg(ivy"com.uber.nullaway:nullaway:0.12.7")
 
   def ivyDeps = super.ivyDeps() ++ Agg(ivy"io.vavr:vavr:0.10.6")
-  
-  def sources = Task.Sources {
-    Seq(PathRef(millSourcePath / "src" / "main" / "java"))
-  }
 
   def compileIvyDeps = super.compileIvyDeps() ++
     Agg(ivy"org.jspecify:jspecify:1.0.0")
 
   def publishVersion = "0.1-SNAPSHOT"
 
-  object test extends JavaTests with TestModule.Junit5 {
+  object test extends MavenTests with TestModule.Junit5 {
 
     def ivyDeps = super.ivyDeps() ++ Agg(
-      ivy"org.junit.jupiter:junit-jupiter:5.12.2",
       ivy"org.mockito:mockito-core:5.18.0",
       ivy"org.mockito:mockito-junit-jupiter:5.18.0"
     )
 
-    def sources = Task.Sources {
-      Seq(PathRef(millSourcePath / "src" / "test" / "java"))
-    }
-
     def testSandboxWorkingDir = false
     def testEnableWorkStealing = false
   }
diff --git a/src/test/java/com/github/tonivade/vavr/effect/ScheduleTest.java b/src/test/java/com/github/tonivade/vavr/effect/ScheduleTest.java
index d8f7da3..4d52349 100644
--- a/src/test/java/com/github/tonivade/vavr/effect/ScheduleTest.java
+++ b/src/test/java/com/github/tonivade/vavr/effect/ScheduleTest.java
@@ -15,7 +15,6 @@
 import java.util.function.Consumer;
 import java.util.function.Supplier;
 
-import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.Mock;
@@ -38,8 +37,6 @@
 
     assertEquals(unit(), result);
     verify(console, times(3)).accept("hola");
-
-    Assertions.fail();
   }
 
   @Test