diff --git a/lib/krane/restart_task.rb b/lib/krane/restart_task.rb
index d729ae12b..9a5e312d5 100644
--- a/lib/krane/restart_task.rb
+++ b/lib/krane/restart_task.rb
@@ -12,10 +12,9 @@ class RestartTask
     class FatalRestartError < FatalDeploymentError; end
 
     class RestartAPIError < FatalRestartError
-      def initialize(deployment_name, response)
+      def initialize(deployment_name, message)
         super("Failed to restart #{deployment_name}. " \
-            "API returned non-200 response code (#{response.code})\n" \
-            "Response:\n#{response.body}")
+            "Error:\n#{message}")
       end
     end