diff --git a/correction.py b/correction.py
index 7551e83..128dbf8 100644
--- a/correction.py
+++ b/correction.py
@@ -237,6 +237,7 @@ completed_set = set((str(f), l) for f, l in completed_tasks)
 tasks_to_process = [t for t in tasks if (str(t[0]), t[1]) not in completed_set]
 
 def process_single_task(task_tuple):
+    global pro_count, flash_count
     file_path, label = task_tuple
     group_name = os.path.splitext(file_path)[0]
     json_path = group_name + '.json'