A bit of fiddling to restore the JS build. (Won't load yet, I presume.)

js/MLS_JS.ml

@@ -74,7 +74,7 @@ let _ =
           Js.null
 
     method currentEpoch (s: MLS.state) =
-      Z.to_int s.MLS.tree_state.MLS_TreeSync_Types.version3
+      Z.to_int s.MLS.epoch
 
     method create1 (e: Typed_array.uint8Array Js.t) (credential: _ Js.t) (priv: Typed_array.uint8Array Js.t)
       (group_id: Js.js_string Js.t)
@@ -152,6 +152,11 @@ let _ =
                     val kind = "add"
                     val identity = js_string_of_bytes identity
                   end)
+              | MLS.MsgRemove identity ->
+                  Obj.magic (object%js
+                    val kind = "remove"
+                    val identity = js_string_of_bytes identity
+                  end)
             val state = state
           end)
       | InternalError s ->

js/dune

@@ -3,9 +3,9 @@
  (modes byte js)
  (modules MLS_JS)
  (js_of_ocaml
-   (flags "--disable" "genprim" "--opt" "3")
+   ;(flags "--disable" "genprim" "--opt" "3")
    ;(flags "--disable" "genprim" "--opt" "3" "--set" "tc_depth=0" "--pretty" "--disable" "excwrap")
    (javascript_files "BigInteger.js" "zarith.js" "overrides.js"))
  (preprocess
   (pps js_of_ocaml-ppx))
- (libraries mls.core js_of_ocaml))
+ (libraries fstar.lib mls_star mls_test js_of_ocaml))

ml/tests/dune

@@ -1,7 +1,15 @@
 (copy_files# src/*)
 
+(library
+ (wrapped false)
+ (libraries mls_star)
+ (name mls_test)
+ (modules :standard \
+   MLS_Test))
+
 (test
- (libraries fstar.lib mls_star yojson)
+ (libraries fstar.lib mls_star mls_test yojson)
+ (modules MLS_Test)
  (name MLS_Test))
 
 (env