diff --git a/run_classifier_pytorch.py b/run_classifier_pytorch.py
index 98527911..eaa32e47 100644
--- a/run_classifier_pytorch.py
+++ b/run_classifier_pytorch.py
@@ -23,6 +23,7 @@ import os
 import logging
 import argparse
 
+import random
 import numpy as np
 from tqdm import tqdm, trange
 import torch
diff --git a/run_squad_pytorch.py b/run_squad_pytorch.py
index a1db682c..701aa229 100644
--- a/run_squad_pytorch.py
+++ b/run_squad_pytorch.py
@@ -26,6 +26,7 @@ import json
 import math
 import os
 from tqdm import tqdm, trange
+import random
 
 import torch
 from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler