diff --git a/examples/distillation/run_squad_w_distillation.py b/examples/distillation/run_squad_w_distillation.py
index 4f58807572..4900f19ead 100644
--- a/examples/distillation/run_squad_w_distillation.py
+++ b/examples/distillation/run_squad_w_distillation.py
@@ -20,6 +20,7 @@ import glob
 import logging
 import os
 import random
+import timeit
 
 import numpy as np
 import torch