diff --git a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp
index 06cc55cdc..55be2f8b7 100755
--- a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp
+++ b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.cpp
@@ -238,9 +238,9 @@ std::shared_ptr<Model> PredictSession::restoreModel(const std::shared_ptr<StepFi
     return model;
 }
 
-std::shared_ptr<Model> PredictSession::restoreModel(int i)
+std::shared_ptr<Model> PredictSession::restoreModel(int i, int skip_mode)
 {
-    return restoreModel(m_stepfiles.at(i));
+    return restoreModel(m_stepfiles.at(i), skip_mode);
 }
 
 // predict one element
diff --git a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h
index 86d5056bd..c9e2575de 100755
--- a/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h
+++ b/lib/smurff-cpp/SmurffCpp/Predict/PredictSession.h
@@ -40,7 +40,7 @@ class PredictSession : public ISession
 
 private:
     std::shared_ptr<Model> restoreModel(const std::shared_ptr<StepFile> &, int skip_mode = -1);
-    std::shared_ptr<Model> restoreModel(int i);
+    std::shared_ptr<Model> restoreModel(int i, int skip_mode = -1);
 
 public:
     int    getNumSteps()  const { return m_stepfiles.size(); }