1 files changed, 3 insertions, 1 deletions
diff --git a/modules/gallery/helpers/gallery_task.php b/modules/gallery/helpers/gallery_task.php
index 1ff26c35..95216cf0 100644
--- a/modules/gallery/helpers/gallery_task.php
+++ b/modules/gallery/helpers/gallery_task.php
@@ -52,7 +52,8 @@ class gallery_task_Core {
       $mode = $task->get("mode", "init");
       if ($mode == "init") {
         $task->set("total_count", $total_count);
-        $task->get("mode", "process");
+        $task->set("mode", "process");
+        batch::start();
       }
 
       $completed = $task->get("completed", 0);
@@ -101,6 +102,7 @@ class gallery_task_Core {
       if ($task->percent_complete == 100) {
         $task->done = true;
         $task->state = "success";
+        batch::stop();
         site_status::clear("graphics_dirty");
       }
     } catch (Exception $e) {