1 files changed, 5 insertions, 1 deletions
diff --git a/klm/util/ersatz_progress.hh b/klm/util/ersatz_progress.hh
index ea6c3bb9..92c345fe 100644
--- a/klm/util/ersatz_progress.hh
+++ b/klm/util/ersatz_progress.hh
@@ -19,7 +19,7 @@ class ErsatzProgress {
     ~ErsatzProgress();
 
     ErsatzProgress &operator++() {
-      if (++current_ == next_) Milestone();
+      if (++current_ >= next_) Milestone();
       return *this;
     }
 
@@ -33,6 +33,10 @@ class ErsatzProgress {
       Milestone();
     }
 
+    void Finished() {
+      Set(complete_);
+    }
+
   private:
     void Milestone();