diff --git a/epi_judge_python/smallest_subarray_covering_all_values.py b/epi_judge_python/smallest_subarray_covering_all_values.py
index b7dc2a595..b768015d0 100644
--- a/epi_judge_python/smallest_subarray_covering_all_values.py
+++ b/epi_judge_python/smallest_subarray_covering_all_values.py
@@ -3,6 +3,7 @@
 from typing import List
 
 from test_framework import generic_test
+from test_framework.test_failure import TestFailure
 from test_framework.test_utils import enable_executor_hook
 
 Subarray = collections.namedtuple('Subarray', ('start', 'end'))