diff --git a/api/tests/unit_tests/controllers/console/test_files.py b/api/tests/unit_tests/controllers/console/test_files.py
index f6ef1cb8241..e62af24c082 100644
--- a/api/tests/unit_tests/controllers/console/test_files.py
+++ b/api/tests/unit_tests/controllers/console/test_files.py
@@ -162,6 +162,7 @@ class TestFileApiPost:
         mock_file.preview_url = "http://example.com/preview/file-id-123"
         mock_file.source_url = "http://example.com/source/file-id-123"
         mock_file.original_url = None
+        mock_file.reference = None
         mock_file.user_id = "user-123"
         mock_file.tenant_id = "tenant-123"
         mock_file.conversation_id = None
@@ -198,6 +199,7 @@ class TestFileApiPost:
         mock_file.preview_url = None
         mock_file.source_url = None
         mock_file.original_url = None
+        mock_file.reference = None
         mock_file.user_id = "user-456"
         mock_file.tenant_id = "tenant-456"
         mock_file.conversation_id = None
diff --git a/api/tests/unit_tests/controllers/service_api/app/test_file.py b/api/tests/unit_tests/controllers/service_api/app/test_file.py
index 88ebe955a8c..9ec4711dbc2 100644
--- a/api/tests/unit_tests/controllers/service_api/app/test_file.py
+++ b/api/tests/unit_tests/controllers/service_api/app/test_file.py
@@ -259,6 +259,7 @@ class TestFileApiPost:
         mock_upload.preview_url = None
         mock_upload.source_url = None
         mock_upload.original_url = None
+        mock_upload.reference = None
         mock_upload.user_id = None
         mock_upload.tenant_id = None
         mock_upload.conversation_id = None