diff --git a/examples/example_image/main.py b/examples/example_image/main.py
index e96094a..485b76b 100644
--- a/examples/example_image/main.py
+++ b/examples/example_image/main.py
@@ -200,4 +200,4 @@ def test_search(gamma=0.1, load_market=True):
 if __name__ == "__main__":
     # prepare_data()
     # prepare_model()
-    test_search(load_market=True)
+    test_search(load_market=False)
diff --git a/examples/example_m5/example_init.py b/examples/example_m5/example_init.py
index c5d26d1..e0aabdd 100644
--- a/examples/example_m5/example_init.py
+++ b/examples/example_m5/example_init.py
@@ -1,6 +1,7 @@
 import os
 import joblib
 import numpy as np
+import lightgbm as lgb
 from learnware.model import BaseModel
 
 
@@ -8,7 +9,7 @@ class Model(BaseModel):
     def __init__(self):
         super(Model, self).__init__(input_shape=(82,), output_shape=())
         dir_path = os.path.dirname(os.path.abspath(__file__))
-        self.model = joblib.load(os.path.join(dir_path, "model.out"))
+        self.model = lgb.Booster(model_file=os.path.join(dir_path, "model.out"))
 
     def fit(self, X: np.ndarray, y: np.ndarray):
         pass
diff --git a/examples/example_pfs/example_init.py b/examples/example_pfs/example_init.py
index d875d96..88b788a 100644
--- a/examples/example_pfs/example_init.py
+++ b/examples/example_pfs/example_init.py
@@ -6,6 +6,7 @@ from learnware.model import BaseModel
 
 class Model(BaseModel):
     def __init__(self):
+        super(Model, self).__init__(input_shape=(31,), output_shape=())
         dir_path = os.path.dirname(os.path.abspath(__file__))
         self.model = joblib.load(os.path.join(dir_path, "model.out"))