diff --git a/src/Dev/SapphireTest.php b/src/Dev/SapphireTest.php
index 6f2f0f8ad..b8ae2fb48 100644
--- a/src/Dev/SapphireTest.php
+++ b/src/Dev/SapphireTest.php
@@ -537,8 +537,8 @@ class SapphireTest extends PHPUnit_Framework_TestCase implements TestOnly
      * @param string $from
      * @param string $subject
      * @param string $content
-     * @return array Contains keys: 'type', 'to', 'from', 'subject','content', 'plainContent', 'attachedFiles',
-     *               'customHeaders', 'htmlContent', 'inlineImages'
+     * @return array|null Contains keys: 'Type', 'To', 'From', 'Subject', 'Content', 'PlainContent', 'AttachedFiles',
+     *               'HtmlContent'
      */
     public function findEmail($to, $from = null, $subject = null, $content = null)
     {
diff --git a/src/Dev/TestMailer.php b/src/Dev/TestMailer.php
index 08567bc3a..d3141b524 100644
--- a/src/Dev/TestMailer.php
+++ b/src/Dev/TestMailer.php
@@ -88,8 +88,8 @@ class TestMailer implements Mailer
      * @param string $from
      * @param string $subject
      * @param string $content
-     * @return array Contains the keys: 'type', 'to', 'from', 'subject', 'content', 'plainContent', 'attachedFiles',
-     *               'customHeaders', 'htmlContent', 'inlineImages'
+     * @return array|null Contains keys: 'Type', 'To', 'From', 'Subject', 'Content', 'PlainContent', 'AttachedFiles',
+     *               'HtmlContent'
      */
     public function findEmail($to, $from = null, $subject = null, $content = null)
     {
@@ -103,7 +103,7 @@ class TestMailer implements Mailer
         foreach ($this->emailsSent as $email) {
             $matched = true;
 
-            foreach (array('To','From','Subject','Content') as $field) {
+            foreach (array('To', 'From', 'Subject', 'Content') as $field) {
                 if ($value = $compare[$field]) {
                     if ($value[0] == '/') {
                         $matched = preg_match($value, $email[$field]);