diff --git a/tests/test_run.py b/tests/test_run.py
index 3bf50eb..8eac091 100644
--- a/tests/test_run.py
+++ b/tests/test_run.py
@@ -32,17 +32,20 @@ async def test_run(async_flag, record_mode):
         output = await replicate.async_run(
             f"stability-ai/sdxl:{version}",
             input=input,
+            use_file_output=True,
         )
     else:
         output = replicate.run(
             f"stability-ai/sdxl:{version}",
             input=input,
+            use_file_output=True,
         )
 
     assert output is not None
     assert isinstance(output, list)
     assert len(output) > 0
-    assert output[0].startswith("https://")
+    assert isinstance(output[0], FileOutput)
+    assert output[0].url.startswith("https://")
 
 
 @pytest.mark.vcr("run__concurrently.yaml")