diff --git a/submission_runner.py b/submission_runner.py
index b0981d941..c94778bbe 100644
--- a/submission_runner.py
+++ b/submission_runner.py
@@ -22,6 +22,7 @@
 import struct
 import time
 from typing import Any, Dict, Optional, Tuple
+from pickle import dump
 
 from absl import app
 from absl import flags