diff --git a/076-077-078-Unet_nuclei_tutorial.py b/076-077-078-Unet_nuclei_tutorial.py
index 5ce0ec3d0..4d97005e4 100644
--- a/076-077-078-Unet_nuclei_tutorial.py
+++ b/076-077-078-Unet_nuclei_tutorial.py
@@ -142,7 +142,8 @@
 
 callbacks = [
         tf.keras.callbacks.EarlyStopping(patience=2, monitor='val_loss'),
-        tf.keras.callbacks.TensorBoard(log_dir='logs')]
+        tf.keras.callbacks.TensorBoard(log_dir='logs'),
+        checkpointer]
 
 results = model.fit(X_train, Y_train, validation_split=0.1, batch_size=16, epochs=25, callbacks=callbacks)