diff --git a/modules_forge/main_thread.py b/modules_forge/main_thread.py
index f0ed6cc8..eb3e7889 100644
--- a/modules_forge/main_thread.py
+++ b/modules_forge/main_thread.py
@@ -26,10 +26,10 @@ class Task:
 
     def work(self):
         global last_exception
-        self.exception = None
-        last_exception = None
         try:
             self.result = self.func(*self.args, **self.kwargs)
+            self.exception = None
+            last_exception = None
         except Exception as e:
             traceback.print_exc()
             print(e)