diff --git a/tests/ai/beta/vision/test_cast.py b/tests/ai/beta/vision/test_cast.py
index fca6449f9..ce3137cb5 100644
--- a/tests/ai/beta/vision/test_cast.py
+++ b/tests/ai/beta/vision/test_cast.py
@@ -112,6 +112,7 @@ def test_map(self):
             Location(city="Washington", state="D.C."),
         )
 
+    @pytest.mark.flaky(reruns=3)
     async def test_async_map(self):
         ny = marvin.beta.Image(
             "https://images.unsplash.com/photo-1568515387631-8b650bbcdb90"
diff --git a/tests/ai/beta/vision/test_extract.py b/tests/ai/beta/vision/test_extract.py
index b4d917d32..d148a67da 100644
--- a/tests/ai/beta/vision/test_extract.py
+++ b/tests/ai/beta/vision/test_extract.py
@@ -57,6 +57,7 @@ def test_ny_image_and_text(self):
             [Location(city="New York City", state="NY")],
         )
 
+    @pytest.mark.flaky(max_runs=3)
     def test_dog(self):
         class Animal(BaseModel, frozen=True):
             type: Literal["cat", "dog", "bird", "frog", "horse", "pig"]
diff --git a/tests/ai/test_cast.py b/tests/ai/test_cast.py
index 3cb5079a8..141df9ea8 100644
--- a/tests/ai/test_cast.py
+++ b/tests/ai/test_cast.py
@@ -95,7 +95,7 @@ def test_cast_text_with_subtle_instructions(self, gpt_4):
 
         def test_str_target_if_only_instructions_provided(self):
             result = marvin.cast(
-                "one", instructions="the numerical representation of the word "
+                "one", instructions="the arabic numeral for the provided word"
             )
             assert isinstance(result, str)
             assert result == "1"