diff --git a/examples/cifar100/run_client.py b/examples/cifar100/run_client.py
index e94a9cbed..caf6c6cb6 100644
--- a/examples/cifar100/run_client.py
+++ b/examples/cifar100/run_client.py
@@ -26,7 +26,7 @@ def get_api_url(api_url: str, api_port: int, secure: bool = False):
     return url
 
 
-def on_train(in_model):
+def on_train(in_model, client_settings):
     # Save model to temp file
     inpath = helper.get_tmp_path()
     with open(inpath, "wb") as fh: