diff --git a/rabit-learn/kmeans/.gitignore b/rabit-learn/kmeans/.gitignore
new file mode 100644
index 000000000..5f8241b66
--- /dev/null
+++ b/rabit-learn/kmeans/.gitignore
@@ -0,0 +1,2 @@
+kmeans
+*.mpi
diff --git a/src/allreduce_mock.h b/src/allreduce_mock.h
index 96bc55800..1e08d57ed 100644
--- a/src/allreduce_mock.h
+++ b/src/allreduce_mock.h
@@ -53,6 +53,11 @@ class AllreduceMock : public AllreduceRobust {
     this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "CheckPoint");
     AllreduceRobust::CheckPoint(global_model, local_model);
   }
+
+  virtual void LazyCheckPoint(const ISerializable *global_model) {
+    this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "LazyCheckPoint");
+    AllreduceRobust::LazyCheckPoint(global_model);
+  }
   
  private:
   // key to identify the mock stage