diff --git a/src/callbacks/model_checkpoint.py b/src/callbacks/model_checkpoint.py
index 1e9b8e6..7a5df2c 100644
--- a/src/callbacks/model_checkpoint.py
+++ b/src/callbacks/model_checkpoint.py
@@ -3,7 +3,6 @@ from typing import Optional, Dict, Any
 
 import lightning.pytorch as pl
 from lightning.pytorch.callbacks.model_checkpoint import ModelCheckpoint
-from soupsieve.util import lower
 
 
 class CheckpointHook(ModelCheckpoint):
diff --git a/src/callbacks/save_images.py b/src/callbacks/save_images.py
index c6cd32b..85e9c0b 100644
--- a/src/callbacks/save_images.py
+++ b/src/callbacks/save_images.py
@@ -15,7 +15,7 @@ def process_fn(image, path):
     Image.fromarray(image).save(path)
 
 class SaveImagesHook(Callback):
-    def __init__(self, save_dir="val", max_save_num=0, compressed=True):
+    def __init__(self, save_dir="val", max_save_num=100, compressed=True):
        self.save_dir = save_dir
        self.max_save_num = max_save_num
        self.compressed = compressed