diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py
index 65715f98dcb2..b41209f7db82 100644
--- a/tests/test_modeling_tf_common.py
+++ b/tests/test_modeling_tf_common.py
@@ -272,7 +272,7 @@ def test_saved_model_with_attentions_output(self):
                 else:
                     output = outputs["attentions"]
 
-                self.assertEqual(len(output), num_out)
+                self.assertEqual(len(outputs), num_out)
                 self.assertEqual(len(output), self.model_tester.num_hidden_layers)
                 self.assertListEqual(
                     list(output[0].shape[-3:]),