diff --git a/impl/persistence/api/src/main/java/io/serverlessworkflow/impl/persistence/PersistenceApplicationBuilder.java b/impl/persistence/api/src/main/java/io/serverlessworkflow/impl/persistence/PersistenceApplicationBuilder.java
index 462cf6ca4..a0e6c8450 100644
--- a/impl/persistence/api/src/main/java/io/serverlessworkflow/impl/persistence/PersistenceApplicationBuilder.java
+++ b/impl/persistence/api/src/main/java/io/serverlessworkflow/impl/persistence/PersistenceApplicationBuilder.java
@@ -25,16 +25,14 @@ public static PersistenceApplicationBuilder builder(
     return new PersistenceApplicationBuilder(builder, writer);
   }
 
-  private final PersistenceInstanceWriter writer;
   private final WorkflowApplication.Builder appBuilder;
 
   protected PersistenceApplicationBuilder(Builder appBuilder, PersistenceInstanceWriter writer) {
     this.appBuilder = appBuilder;
-    this.writer = writer;
+    appBuilder.withListener(new WorkflowPersistenceListener(writer));
   }
 
   public WorkflowApplication build() {
-    appBuilder.withListener(new WorkflowPersistenceListener(writer));
     return appBuilder.build();
   }
 }