diff --git a/CMakeLists.txt b/CMakeLists.txt
index 771dcc8..0b5fcf8 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.25)
-project(COSC-4P80-Final-Project VERSION 0.0.16)
+project(COSC-4P80-Final-Project VERSION 0.0.17)
 
 option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
 option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
diff --git a/src/MNIST.cpp b/src/MNIST.cpp
index 63833a0..8e7047c 100644
--- a/src/MNIST.cpp
+++ b/src/MNIST.cpp
@@ -617,7 +617,8 @@ namespace fp
             std::exit(-1);
         }
 
-        state << i;
+        // user can skip this if required.
+        state << i - 1;
         state << '\n';
         state << last_epoch_save;
         state << '\n';