diff --git a/pyinfra/queue/callback.py b/pyinfra/queue/callback.py
index c1470af..39d40d1 100644
--- a/pyinfra/queue/callback.py
+++ b/pyinfra/queue/callback.py
@@ -1,4 +1,4 @@
-from typing import Callable, Union
+from typing import Callable
 
 from dynaconf import Dynaconf
 from kn_utils.logging import logger
@@ -7,9 +7,10 @@ from pyinfra.storage.connection import get_storage
 from pyinfra.storage.utils import (
     download_data_bytes_as_specified_in_message,
     upload_data_as_specified_in_message,
+    DownloadedData,
 )
 
-DataProcessor = Callable[[Union[dict, bytes], dict], Union[dict, list, str]]
+DataProcessor = Callable[[dict[str, DownloadedData] | DownloadedData, dict], dict | list | str]
 Callback = Callable[[dict], dict]
 
 
@@ -28,7 +29,9 @@ def make_download_process_upload_callback(data_processor: DataProcessor, setting
 
         storage = get_storage(settings, queue_message_payload.get("X-TENANT-ID"))
 
-        data: dict[str, bytes] | bytes = download_data_bytes_as_specified_in_message(storage, queue_message_payload)
+        data: dict[str, DownloadedData] | DownloadedData = download_data_bytes_as_specified_in_message(
+            storage, queue_message_payload
+        )
 
         result = data_processor(data, queue_message_payload)
 
diff --git a/pyproject.toml b/pyproject.toml
index 6fffe24..351fa38 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyinfra"
-version = "3.4.2"
+version = "4.0.0"
 description = ""
 authors = ["Team Research <research@knecon.com>"]
 license = "All rights reseverd"