diff --git a/autogpt_platform/backend/backend/blocks/test/test_blocks_dos_vulnerability.py b/autogpt_platform/backend/backend/blocks/test/test_blocks_dos_vulnerability.py
index e724756882..e2e44b194c 100644
--- a/autogpt_platform/backend/backend/blocks/test/test_blocks_dos_vulnerability.py
+++ b/autogpt_platform/backend/backend/blocks/test/test_blocks_dos_vulnerability.py
@@ -239,6 +239,7 @@ class TestStoreMediaFileSecurity:
                     user_id="test_user",
                     graph_exec_id="test",
                 ),
+                return_format="for_local_processing",
             )
 
     @patch("backend.util.file.Path")
@@ -280,4 +281,5 @@ class TestStoreMediaFileSecurity:
                     user_id="test_user",
                     graph_exec_id="test",
                 ),
+                return_format="for_local_processing",
             )
diff --git a/autogpt_platform/backend/backend/util/file_test.py b/autogpt_platform/backend/backend/util/file_test.py
index d75a786c76..9fe672d155 100644
--- a/autogpt_platform/backend/backend/util/file_test.py
+++ b/autogpt_platform/backend/backend/util/file_test.py
@@ -245,4 +245,5 @@ class TestFileCloudIntegration:
                 await store_media_file(
                     file=MediaFileType(cloud_path),
                     execution_context=make_test_context(graph_exec_id=graph_exec_id),
+                    return_format="for_local_processing",
                 )