1 files changed, 5 insertions, 0 deletions
diff --git a/lib/gitlab/kubernetes/helm/install_command.rb b/lib/gitlab/kubernetes/helm/install_command.rb
index 8e24cb4c24f..3784aecccb5 100644
--- a/lib/gitlab/kubernetes/helm/install_command.rb
+++ b/lib/gitlab/kubernetes/helm/install_command.rb
@@ -44,6 +44,7 @@ module Gitlab
         def install_command
           command = ['helm', 'upgrade', name, chart] +
             install_flag +
+            rollback_support_flag +
             reset_values_flag +
             tls_flags_if_remote_tiller +
             optional_version_flag +
@@ -83,6 +84,10 @@ module Gitlab
 
           ['--version', version]
         end
+
+        def rollback_support_flag
+          ['--atomic', '--cleanup-on-fail']
+        end
       end
     end
   end