diff --git a/sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OnlineModelConfig.java b/sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OnlineModelConfig.java
index 3cff42d9..b67676df 100644
--- a/sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OnlineModelConfig.java
+++ b/sherpa-onnx/java-api/src/com/k2fsa/sherpa/onnx/OnlineModelConfig.java
@@ -140,12 +140,14 @@ public class OnlineModelConfig {
             return this;
         }
 
-        public void setModelingUnit(String modelingUnit) {
+        public Builder setModelingUnit(String modelingUnit) {
             this.modelingUnit = modelingUnit;
+            return this;
         }
 
-        public void setBpeVocab(String bpeVocab) {
+        public Builder setBpeVocab(String bpeVocab) {
             this.bpeVocab = bpeVocab;
+            return this;
         }
     }
 }