diff --git a/rust/e2e/src/test_client/utils.rs b/rust/e2e/src/test_client/utils.rs
index 5a1f8f036..ddedc580f 100644
--- a/rust/e2e/src/test_client/utils.rs
+++ b/rust/e2e/src/test_client/utils.rs
@@ -6,7 +6,13 @@ use xaynet_core::{
     mask::{MaskConfig, Model},
     ParticipantSecretKey,
 };
-use xaynet_sdk::{client::Client as ApiClient, ModelStore, Notify, PetSettings, StateMachine};
+use xaynet_sdk::{
+    client::Client as ApiClient,
+    settings::PetSettings,
+    ModelStore,
+    Notify,
+    StateMachine,
+};
 
 #[derive(Ord, PartialOrd, Eq, PartialEq)]
 pub enum ClientType {
@@ -79,10 +85,11 @@ fn compute_signatures(
 pub struct Notifier;
 
 impl Notify for Notifier {
-    fn notify_new_round(&mut self) {}
-    fn notify_sum(&mut self) {}
-    fn notify_update(&mut self) {}
-    fn notify_idle(&mut self) {}
+    fn new_round(&mut self) {}
+    fn sum(&mut self) {}
+    fn update(&mut self) {}
+    fn idle(&mut self) {}
+    fn load_model(&mut self) {}
 }
 
 pub struct LocalModel(pub Arc<Model>);