diff --git a/Training/train.swift b/Training/train.swift
index 6b2273f..03cf912 100644
--- a/Training/train.swift
+++ b/Training/train.swift
@@ -181,6 +181,9 @@ final class ClassifierCreator {
     }
     
     private func loadImage(cap: Int, image: Int) async -> Bool {
+        guard createFolderIfMissing(imageDirectory.appendingPathComponent(String(format: "%04d", cap))) else {
+            return false
+        }
         let url = imageUrl(base: server.appendingPathComponent("images"), cap: cap, image: image)
         let tempFile: URL, response: URLResponse
         do {