diff --git a/tuner/tuner/candidate_gen_test.py b/tuner/tuner/candidate_gen_test.py
index 6a44ac27e..a1a3a3e49 100644
--- a/tuner/tuner/candidate_gen_test.py
+++ b/tuner/tuner/candidate_gen_test.py
@@ -12,7 +12,7 @@
 from . import candidate_gen
 
 from iree.compiler import ir  # type: ignore
-from iree.compiler.dialects import func # type: ignore
+from iree.compiler.dialects import func  # type: ignore
 
 
 def test_get_shaped_type_element_bitwidth() -> None: