diff --git a/experimental/types/src/main/java/io/serverlessworkflow/api/types/func/UntilPredicate.java b/experimental/types/src/main/java/io/serverlessworkflow/api/types/func/UntilPredicate.java
index 3d576fd1..8af8e46e 100644
--- a/experimental/types/src/main/java/io/serverlessworkflow/api/types/func/UntilPredicate.java
+++ b/experimental/types/src/main/java/io/serverlessworkflow/api/types/func/UntilPredicate.java
@@ -24,13 +24,13 @@ public class UntilPredicate extends Until {
   private Predicate<?> predicate;
   private Optional<Class<?>> predicateClass;
 
-  public <T> Until withFunction(Predicate<T> predicate) {
+  public <T> UntilPredicate withPredicate(Predicate<T> predicate) {
     this.predicate = predicate;
     this.predicateClass = Optional.empty();
     return this;
   }
 
-  public <T> Until withFunction(Predicate<T> predicate, Class<T> clazz) {
+  public <T> UntilPredicate withPredicate(Predicate<T> predicate, Class<T> clazz) {
     this.predicate = predicate;
     this.predicateClass = Optional.ofNullable(clazz);
     return this;