diff --git a/mnist/main.py b/mnist/main.py
index 7d7899d9..66d755fc 100644
--- a/mnist/main.py
+++ b/mnist/main.py
@@ -6,7 +6,7 @@
 import torch.optim as optim
 from torchvision import datasets, transforms
 from torch.optim.lr_scheduler import StepLR
-
+import json
 
 class Net(nn.Module):
     def __init__(self):