diff --git a/tests/test_outline.py b/tests/test_outline.py
index 2c809ce31..9ffe31f69 100644
--- a/tests/test_outline.py
+++ b/tests/test_outline.py
@@ -18,6 +18,7 @@ def test_outline():
     mock_model = Mock()
     mock_model.__iter__ = lambda self: iter([])
     mock_generator = Mock()
+    mock_generator.__iter__ = lambda self: iter([])
     mock_generator.return_value = '{"result": 6}'
     with patch("outlines.generate.json", return_value=mock_generator):
         outline_instance = Outline(mock_model, template, OutputModel)