diff --git a/src/backend/langflow/services/task/backends/base.py b/src/backend/langflow/services/task/backends/base.py
index a05c760e7..ccbd9273b 100644
--- a/src/backend/langflow/services/task/backends/base.py
+++ b/src/backend/langflow/services/task/backends/base.py
@@ -1,14 +1,12 @@
 from abc import ABC, abstractmethod
-from typing import Any, Callable, Union
+from typing import Any, Callable
 
 
 class TaskBackend(ABC):
     @abstractmethod
-    def launch_task(
-        self, task_func: Callable[..., Any], *args: Any, **kwargs: Any
-    ) -> Union[int, str]:
+    def launch_task(self, task_func: Callable[..., Any], *args: Any, **kwargs: Any):
         pass
 
     @abstractmethod
-    def get_task(self, task_id: Union[int, str]) -> Any:
+    def get_task(self, task_id: str) -> Any:
         pass