diff --git a/tests/ai/test_classify.py b/tests/ai/test_classify.py
index 0bef4bcf1..390621d10 100644
--- a/tests/ai/test_classify.py
+++ b/tests/ai/test_classify.py
@@ -123,6 +123,7 @@ def test_return_index(self):
             assert result == 1
 
     class TestExamples:
+        @pytest.mark.flaky(max_runs=3)
         async def test_hogwarts_sorting_hat(self):
             description = "Brave, daring, chivalrous, and sometimes a bit reckless -- just like Harry Potter."
 
diff --git a/tests/ai/vision/test_classify.py b/tests/ai/vision/test_classify.py
index 2248ef574..d4d8fece5 100644
--- a/tests/ai/vision/test_classify.py
+++ b/tests/ai/vision/test_classify.py
@@ -21,8 +21,7 @@ def test_ny_image_and_text(self):
             "https://images.unsplash.com/photo-1568515387631-8b650bbcdb90"
         )
         result = marvin.classify(
-            data="I see the empire state building",
-            images=[img],
+            data=["I see a tall building", img],
             labels=["urban", "rural"],
         )
         assert result == "urban"
diff --git a/tests/ai/vision/test_extract.py b/tests/ai/vision/test_extract.py
index 0dc2e46d0..01202340c 100644
--- a/tests/ai/vision/test_extract.py
+++ b/tests/ai/vision/test_extract.py
@@ -82,7 +82,7 @@ def test_dog_breeds_with_text(self):
             "https://images.unsplash.com/photo-1548199973-03cce0bbc87b?q=80&w=2969&auto=format&fit=crop&ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D"
         )
         result = marvin.extract(
-            [img, "collie"],
+            [img, "Another type is: collie"],
             target=str,
             instructions="extract all dog breeds pictured or mentioned",
         )