diff --git a/MNIST_99.4/check_model.py b/MNIST_99.4/check_model.py
index 7668d37..3de308b 100644
--- a/MNIST_99.4/check_model.py
+++ b/MNIST_99.4/check_model.py
@@ -1,5 +1,5 @@
-from models.model import MNISTModel
+from models.model import FastMNIST
 from utils import print_model_summary
 
-model = MNISTModel()
+model = FastMNIST()
 print_model_summary(model)