diff --git a/tests/recipes/wrangles/test_extract.py b/tests/recipes/wrangles/test_extract.py
index 49ceda8c..4904b526 100644
--- a/tests/recipes/wrangles/test_extract.py
+++ b/tests/recipes/wrangles/test_extract.py
@@ -1873,20 +1873,24 @@ def test_extract_custom_ai_multiple_output(self):
         Test extract.custom with AI that produces
         a multi column output
         """
-        df = wrangles.recipe.run(
-            """
-            read:
-            - test:
-                rows: 1
-                values:
-                    header: example 1 2 3 4 5 word
-            wrangles:
-            - extract.custom:
-                input: header
-                output: results
-                model_id: 1f3ba62b-ce20-486e
-            """
-        )
+        with patch(
+            "wrangles.extract.custom",
+            return_value=[{"Words": ["word"], "Numbers": ["1", "2", "3", "4", "5"]}]
+        ):
+            df = wrangles.recipe.run(
+                """
+                read:
+                - test:
+                    rows: 1
+                    values:
+                        header: example 1 2 3 4 5 word
+                wrangles:
+                - extract.custom:
+                    input: header
+                    output: results
+                    model_id: 1f3ba62b-ce20-486e
+                """
+            )
         assert (
             "Words" in df["results"][0] and
             "Numbers" in df["results"][0]