diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py
index de6b1dcac..9e4103ecd 100644
--- a/tests/e2e/integrations/test_kd.py
+++ b/tests/e2e/integrations/test_kd.py
@@ -64,6 +64,7 @@ def min_cfg(temp_dir):
         "max_steps": 5,
         "output_dir": temp_dir,
         "save_safetensors": True,
+        "use_tensorboard": True,
     }