diff --git a/tests/test_functional_text_generation_example.py b/tests/test_functional_text_generation_example.py
index 9012cabd65..8b57847d17 100644
--- a/tests/test_functional_text_generation_example.py
+++ b/tests/test_functional_text_generation_example.py
@@ -22,7 +22,7 @@
     }
 else:
     # Functional testing only on G2 onwards
-    MODEL_OUTPUTS = []
+    MODEL_OUTPUTS = {}
 
 
 def _test_text_generation(