diff --git a/test/t1.ml b/test/t1.ml
index 87e4792..b96a81f 100644
--- a/test/t1.ml
+++ b/test/t1.ml
@@ -25,9 +25,9 @@ let run () =
           Trace.enter_manual_sub_span ~parent:pseudo_async_sp
             ~flavor:
               (if _i mod 3 = 0 then
-                `Sync
-              else
-                `Async)
+                 `Sync
+               else
+                 `Async)
             ~__FILE__ ~__LINE__ "sub-sleep"
         in
 
@@ -43,5 +43,5 @@ let run () =
   done
 
 let () =
-  Trace_tef.Internal_.mock_all_ ();
+  Trace_tef.Private_.mock_all_ ();
   Trace_tef.with_setup ~out:`Stdout () @@ fun () -> run ()
diff --git a/test/t2.ml b/test/t2.ml
index 1ad9de9..e2868c9 100644
--- a/test/t2.ml
+++ b/test/t2.ml
@@ -14,7 +14,7 @@ let%trace rec fib2 x =
     fib2 (x - 1) + fib2 (x - 2)
 
 let () =
-  Trace_tef.Internal_.mock_all_ ();
+  Trace_tef.Private_.mock_all_ ();
   let@ () = Trace_tef.with_setup ~out:`Stdout () in
 
   Trace_core.set_process_name "main";