diff --git a/tests/models/bloom/test_modeling_flax_bloom.py b/tests/models/bloom/test_modeling_flax_bloom.py
index 0e49039afe59..91716615de9a 100644
--- a/tests/models/bloom/test_modeling_flax_bloom.py
+++ b/tests/models/bloom/test_modeling_flax_bloom.py
@@ -198,7 +198,7 @@ def test_model_from_pretrained(self):
 @slow
 @require_flax
 class FlaxBloomGenerationTest(unittest.TestCase):
-    all_model_classes = (FlaxBloomForCausalLM) if is_flax_available() else ()
+    all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else ()
     all_generative_model_classes = () if is_flax_available() else ()
 
     def setUp(self):