diff --git a/go.mod b/go.mod
index 44b1216..7df562d 100644
--- a/go.mod
+++ b/go.mod
@@ -4,7 +4,7 @@ go 1.18
 
 require (
 	github.com/stretchr/testify v1.7.2
-	github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230807124901-05ef8aca570d
+	github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230821155521-70f1ff564bae
 	github.com/waku-org/go-zerokit-rln-arm v0.0.0-20230807124929-ea702b1b4305
 	github.com/waku-org/go-zerokit-rln-x86_64 v0.0.0-20230807124913-ea636e5b4005
 )
diff --git a/go.sum b/go.sum
index f2abfe3..a04ea0f 100644
--- a/go.sum
+++ b/go.sum
@@ -17,6 +17,8 @@ github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230803113401-9a7ef94d120e h1:A
 github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230803113401-9a7ef94d120e/go.mod h1:KYykqtdApHVYZ3G0spwMnoxc5jH5eI3jyO9SwsSfi48=
 github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230807124901-05ef8aca570d h1:7Ot7vJAniJWQkarZBFxmjRo39gGksKcqs4kZ10l+szs=
 github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230807124901-05ef8aca570d/go.mod h1:KYykqtdApHVYZ3G0spwMnoxc5jH5eI3jyO9SwsSfi48=
+github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230821155521-70f1ff564bae h1:VXgstV6RFUs6L/x0Xad4s0BIJ8hVEv1pyrByYzvZdT0=
+github.com/waku-org/go-zerokit-rln-apple v0.0.0-20230821155521-70f1ff564bae/go.mod h1:KYykqtdApHVYZ3G0spwMnoxc5jH5eI3jyO9SwsSfi48=
 github.com/waku-org/go-zerokit-rln-arm v0.0.0-20230801152407-8101ff87ee0a h1:10cre+P76QvnLeyeCVAM8WDbUCri/y5xY3LtwI9Y5DE=
 github.com/waku-org/go-zerokit-rln-arm v0.0.0-20230801152407-8101ff87ee0a/go.mod h1:7cSGUoGVIla1IpnChrLbkVjkYgdOcr7rcifEfh4ReR4=
 github.com/waku-org/go-zerokit-rln-arm v0.0.0-20230804193517-1b9eff57e592 h1:v1+rpINORPzMfBxoLpJMTB+8Z69gb7tbfTu7gAn4heg=
diff --git a/rln/link/apple.go b/rln/link/apple.go
index 3c1cfb5..c1ef8b3 100644
--- a/rln/link/apple.go
+++ b/rln/link/apple.go
@@ -11,16 +11,16 @@ type RLNWrapper struct {
 	ffi *r.RLN
 }
 
-func NewWithParams(depth int, wasm []byte, zkey []byte, verifKey []byte) (*RLNWrapper, error) {
-	rln, err := r.NewWithParams(depth, wasm, zkey, verifKey)
+func NewWithParams(depth int, wasm []byte, zkey []byte, verifKey []byte, treeConfig []byte) (*RLNWrapper, error) {
+	rln, err := r.NewWithParams(depth, wasm, zkey, verifKey, treeConfig)
 	if err != nil {
 		return nil, err
 	}
 	return &RLNWrapper{ffi: rln}, nil
 }
 
-func NewWithFolder(depth int, resourcesFolderPath string) (*RLNWrapper, error) {
-	rln, err := r.NewWithFolder(depth, resourcesFolderPath)
+func New(depth int, config []byte) (*RLNWrapper, error) {
+	rln, err := r.New(uint(depth), config)
 	if err != nil {
 		return nil, err
 	}
@@ -106,3 +106,7 @@ func (i RLNWrapper) SetMetadata(metadata []byte) bool {
 func (i RLNWrapper) GetMetadata() ([]byte, error) {
 	return i.ffi.GetMetadata()
 }
+
+func (i RLNWrapper) Flush() bool {
+	return i.ffi.Flush()
+}