diff --git a/tests/test_hf_mpt_gen.py b/tests/test_hf_mpt_gen.py
index 4714739170..a2c5bf7f98 100644
--- a/tests/test_hf_mpt_gen.py
+++ b/tests/test_hf_mpt_gen.py
@@ -183,4 +183,4 @@ def test_mpt_generate_callback(callback_generate: Any, tmpdir: Path):
     trainer.logger.log_table = Mock()
     trainer.fit()
 
-    assert callback_generate.assert_called_once
+    callback_generate.assert_called_once()