diff --git a/Training/train.swift b/Training/train.swift
index 37b114d..6b2273f 100644
--- a/Training/train.swift
+++ b/Training/train.swift
@@ -144,6 +144,7 @@ final class ClassifierCreator {
             return nil
         }
         let changedImageList = await getChangedImageList()
+            .filter { $0.image < imageCounts[$0.cap] ?? 0 } // Filter non-existent images
         
         if changedImageList.isEmpty {
             print("[INFO] No changed images to load")