diff --git a/neuralnetlib/model.py b/neuralnetlib/model.py
index ba2778f..79b4713 100644
--- a/neuralnetlib/model.py
+++ b/neuralnetlib/model.py
@@ -186,13 +186,13 @@ def fit(self, x_train: np.ndarray, y_train: np.ndarray, epochs: int, batch_size:
                         break
                     if callback.on_epoch_end(self, metrics_values):
                         break
+                    
+                if any(callback.stop_training for callback in callbacks):
+                                break
 
             if verbose:
                 print()
 
-            if any(callback.stop_training for callback in callbacks):
-                break
-
         if verbose:
             print()
 
diff --git a/setup.py b/setup.py
index 85ed1e1..088067c 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='neuralnetlib',
-    version='2.3.3',
+    version='2.3.4',
     author='Marc Pinet',
     description='A simple convolutional neural network library with only numpy as dependency',
     long_description=open('README.md', encoding="utf-8").read(),