diff --git a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_pubmedqa.py b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_pubmedqa.py
index eea2220d..3894d066 100644
--- a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_pubmedqa.py
+++ b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_pubmedqa.py
@@ -25,7 +25,7 @@
 # pylint: disable=g-import-not-at-top
 with konfig.imports():
   from gemma.diffusion import _models
-  from gemma.diffusion import _paths  # pytlint: disable=unused-import
+  from gemma.diffusion import _paths  # pylint: disable=unused-import
   from gemma.diffusion.hackable_diffusion_adapter.data.pubmedqa import pubmedqa_data
   from gemma.diffusion.hackable_diffusion_adapter.eval import pubmedqa_eval
   from gemma.diffusion.hackable_diffusion_adapter.hd import gemma_checkpointer
diff --git a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku.py b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku.py
index 9f00d30e..68d0b2f8 100644
--- a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku.py
+++ b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku.py
@@ -24,7 +24,7 @@
 # pylint: disable=g-import-not-at-top
 with konfig.imports():
   from gemma.diffusion import _models
-  from gemma.diffusion import _paths  # pytlint: disable=unused-import
+  from gemma.diffusion import _paths  # pylint: disable=unused-import
   from gemma.diffusion.hackable_diffusion_adapter.data.sudoku import sudoku_data
   from gemma.diffusion.hackable_diffusion_adapter.eval import sudoku_eval
   from gemma.diffusion.hackable_diffusion_adapter.hd import gemma_checkpointer
diff --git a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku_full.py b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku_full.py
index 4fd00dcc..11821bc4 100644
--- a/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku_full.py
+++ b/gemma/diffusion/hackable_diffusion_adapter/configs/sft_sudoku_full.py
@@ -83,7 +83,7 @@ def new_call(
 # pylint: disable=g-import-not-at-top
 with konfig.imports():
   from gemma.diffusion import _models
-  from gemma.diffusion import _paths  # pytlint: disable=unused-import
+  from gemma.diffusion import _paths  # pylint: disable=unused-import
   from gemma.diffusion.hackable_diffusion_adapter.data.sudoku import sudoku_data
   from gemma.diffusion.hackable_diffusion_adapter.eval import sudoku_eval
   from gemma.diffusion.hackable_diffusion_adapter.hd import gemma_checkpointer