diff --git a/run_squad.py b/run_squad.py
index fa7575de2..46c689496 100644
--- a/run_squad.py
+++ b/run_squad.py
@@ -28,6 +28,7 @@ import os
 from tqdm import tqdm, trange
 import random
 
+import numpy as np
 import torch
 from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
 from torch.utils.data.distributed import DistributedSampler