diff --git a/tests/e2e/test_dpo.py b/tests/e2e/test_dpo.py
index 156949fd8..db1e0f117 100644
--- a/tests/e2e/test_dpo.py
+++ b/tests/e2e/test_dpo.py
@@ -116,7 +116,7 @@ class TestDPOLlamaLora(unittest.TestCase):
         assert (Path(temp_dir) / "checkpoint-20/adapter_model.safetensors").exists()
 
     @with_temp_dir
-    def test_dpo_nll_use_weighting(self, temp_dir):
+    def test_dpo_use_weighting(self, temp_dir):
         # pylint: disable=duplicate-code
         cfg = DictDefault(
             {