diff --git a/app/services/create_my_module_service.rb b/app/services/create_my_module_service.rb
index bcad2d0dd..7c1997891 100644
--- a/app/services/create_my_module_service.rb
+++ b/app/services/create_my_module_service.rb
@@ -28,7 +28,7 @@ class CreateMyModuleService
 
       @my_module.save!
       create_my_module_activity
-      @experiment.generate_workflow_img
+      @params[:experiment].generate_workflow_img
       @my_module
     end
   end