diff --git a/test/util.h b/test/util.h
index 5cbcf241..6aa1a6b9 100644
--- a/test/util.h
+++ b/test/util.h
@@ -83,7 +83,7 @@ class MockFSM : public braft::StateMachine {
             _on_leader_start_closure = NULL;
         }
     }
-    void on_leader_stop(const braft::LeaderChangeContext&) {
+    void on_leader_stop(const butil::Status&) {
         _leader_term = -1;
     }