diff --git a/otter/grade/containers.py b/otter/grade/containers.py
index 809ebf2d..1e2ccbf8 100644
--- a/otter/grade/containers.py
+++ b/otter/grade/containers.py
@@ -7,7 +7,6 @@
 import shutil
 import tempfile
 import zipfile
-from ..test_files import GradingResults
 
 from concurrent.futures import ThreadPoolExecutor, wait
 from python_on_whales import docker
@@ -17,6 +16,7 @@
 from .utils import OTTER_DOCKER_IMAGE_NAME, merge_scores_to_df, TimeoutException
 
 from ..run.run_autograder.autograder_config import AutograderConfig
+from ..test_files import GradingResults
 from ..utils import loggers, OTTER_CONFIG_FILENAME