diff --git a/test/unit/custom_workflow_mailer_test.rb b/test/unit/custom_workflow_mailer_test.rb
index 964d915..551a10b 100644
--- a/test/unit/custom_workflow_mailer_test.rb
+++ b/test/unit/custom_workflow_mailer_test.rb
@@ -42,16 +42,20 @@ def test_truth
   def test_custom_email
     CustomWorkflowMailer.deliver_custom_email @user2, subject: 'Subject', text_body: 'Body', html_body: 'Body'
     email = last_email
-    assert text_part(email).body.include?('Body'), "'Body' expected\n'#{response.body}' present'"
-    assert html_part(email).body.include?('Body'), "'Body' expected\n'#{response.body}' present'"
+    text = text_part(email).body
+    html = html_part(email).body
+    assert text.include?('Body'), "'Body' expected\n'#{text}' present'"
+    assert html.include?('Body'), "'Body' expected\n'#{html}' present'"
   end
 
   def test_custom_email_template
     CustomWorkflowMailer.deliver_custom_email @user2, subject: 'Subject', template_name: 'mailer/test_email',
                                               template_params: { url: Setting.host_name }
     email = last_email
-    assert text_part(email).body.include?('Body'), "'#{Setting.host_name} expected\n'#{response.body}' present'"
-    assert html_part(email).body.include?('Body'), "'#{Setting.host_name} expected\n'#{response.body}' present'"
+    text = text_part(email).body
+    html = html_part(email).body
+    assert text.include?('Body'), "'#{Setting.host_name} expected\n'#{text}' present'"
+    assert html.include?('Body'), "'#{Setting.host_name} expected\n'#{html}' present'"
   end
 
   private