diff --git a/backend/image_generation/core.py b/backend/image_generation/core.py
index dfd3375..91a5535 100644
--- a/backend/image_generation/core.py
+++ b/backend/image_generation/core.py
@@ -27,7 +27,7 @@ async def process_tasks(
 
     processed_results: List[Union[str, None]] = []
     for result in results:
-        if isinstance(result, Exception):
+        if isinstance(result, BaseException):
             print(f"An exception occurred: {result}")
             processed_results.append(None)
         else: