diff --git a/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift b/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift
index ad26a65..18cb217 100644
--- a/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift
+++ b/ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift
@@ -17,7 +17,7 @@ class HandwrittenDigitRecognizer {
             let module = InferenceModule(fileAtPath: filePath) {
             return module
         } else {
-            fatalError("Failed to load model file vit4mnist.pth")
+            fatalError("Failed to load model file vit4mnist.ptl")
         }
     }()