1 files changed, 10 insertions, 4 deletions
diff --git a/app/services/merge_requests/update_service.rb b/app/services/merge_requests/update_service.rb
index 8995c5f2411..b613d88aee4 100644
--- a/app/services/merge_requests/update_service.rb
+++ b/app/services/merge_requests/update_service.rb
@@ -4,7 +4,7 @@ module MergeRequests
   class UpdateService < MergeRequests::BaseService
     extend ::Gitlab::Utils::Override
 
-    def initialize(project, user = nil, params = {})
+    def initialize(project:, current_user: nil, params: {})
       super
 
       @target_branch_was_deleted = @params.delete(:target_branch_was_deleted)
@@ -222,7 +222,7 @@ module MergeRequests
 
     def handle_assignees_change(merge_request, old_assignees)
       MergeRequests::HandleAssigneesChangeService
-        .new(project, current_user)
+        .new(project: project, current_user: current_user)
         .async_execute(merge_request, old_assignees)
     end
 
@@ -295,6 +295,8 @@ module MergeRequests
       case attribute
       when :assignee_ids
         assignees_service.execute(merge_request)
+      when :spend_time
+        add_time_spent_service.execute(merge_request)
       else
         nil
       end
@@ -302,9 +304,13 @@ module MergeRequests
 
     def assignees_service
       @assignees_service ||= ::MergeRequests::UpdateAssigneesService
-        .new(project, current_user, params)
+        .new(project: project, current_user: current_user, params: params)
+    end
+
+    def add_time_spent_service
+      @add_time_spent_service ||= ::MergeRequests::AddSpentTimeService.new(project: project, current_user: current_user, params: params)
     end
   end
 end
 
-MergeRequests::UpdateService.prepend_if_ee('EE::MergeRequests::UpdateService')
+MergeRequests::UpdateService.prepend_mod_with('MergeRequests::UpdateService')