From d5c5c55bf2115e5052bb0e0153db094d38a5f140 Mon Sep 17 00:00:00 2001
From: zmagod <zmago_devetak@yahoo.com>
Date: Tue, 13 Sep 2016 16:32:33 +0200
Subject: [PATCH] refactor

---
 app/models/experiment.rb | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/models/experiment.rb b/app/models/experiment.rb
index 6a6e9d25a..2b24ee6fa 100644
--- a/app/models/experiment.rb
+++ b/app/models/experiment.rb
@@ -499,7 +499,12 @@ class Experiment < ActiveRecord::Base
         raise ActiveRecord::ActiveRecordError
       end
 
-      my_module.save!
+      my_module.save
+    end
+
+    # Generate workflow image for the experiment in which we moved the task
+    to_move.each_value do |id|
+      experiment = Experiment.find_by_id(id)
       experiment.delay.generate_workflow_img
     end
   end
@@ -543,9 +548,14 @@ class Experiment < ActiveRecord::Base
 
         group.experiment = experiment
         group.save!
-        experiment.delay.generate_workflow_img
       end
     end
+
+    # Generate workflow image for the experiment in which we moved the workflow
+    to_move.each_value do |id|
+      experiment = Experiment.find_by_id(id)
+      experiment.delay.generate_workflow_img
+    end
   end
 
   # Update connections for all modules in this project.