diff --git a/src/Mailbox.php b/src/Mailbox.php
index f57e79bf..179c9ed8 100644
--- a/src/Mailbox.php
+++ b/src/Mailbox.php
@@ -123,8 +123,8 @@ public function getStatus(int $flags = null): \stdClass
     /**
      * Bulk Set Flag for Messages.
      *
-     * @param string                       $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
-     * @param array|MessageIterator|string $numbers Message numbers
+     * @param string                                $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
      */
     public function setFlag(string $flag, $numbers): bool
     {
@@ -134,8 +134,8 @@ public function setFlag(string $flag, $numbers): bool
     /**
      * Bulk Clear Flag for Messages.
      *
-     * @param string                       $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
-     * @param array|MessageIterator|string $numbers Message numbers
+     * @param string                                $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
      */
     public function clearFlag(string $flag, $numbers): bool
     {
@@ -277,8 +277,8 @@ public function getThread(): array
     /**
      * Bulk move messages.
      *
-     * @param array|MessageIterator|string $numbers Message numbers
-     * @param MailboxInterface             $mailbox Destination Mailbox to move the messages to
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
+     * @param MailboxInterface                      $mailbox Destination Mailbox to move the messages to
      *
      * @throws \Ddeboer\Imap\Exception\MessageMoveException
      */
@@ -292,8 +292,8 @@ public function move($numbers, MailboxInterface $mailbox): void
     /**
      * Bulk copy messages.
      *
-     * @param array|MessageIterator|string $numbers Message numbers
-     * @param MailboxInterface             $mailbox Destination Mailbox to copy the messages to
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
+     * @param MailboxInterface                      $mailbox Destination Mailbox to copy the messages to
      *
      * @throws \Ddeboer\Imap\Exception\MessageCopyException
      */
diff --git a/src/MailboxInterface.php b/src/MailboxInterface.php
index efe432da..80c32935 100644
--- a/src/MailboxInterface.php
+++ b/src/MailboxInterface.php
@@ -45,16 +45,16 @@ public function getStatus(int $flags = null): \stdClass;
     /**
      * Bulk Set Flag for Messages.
      *
-     * @param string                       $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
-     * @param array|MessageIterator|string $numbers Message numbers
+     * @param string                                $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
      */
     public function setFlag(string $flag, $numbers): bool;
 
     /**
      * Bulk Clear Flag for Messages.
      *
-     * @param string                       $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
-     * @param array|MessageIterator|string $numbers Message numbers
+     * @param string                                $flag    \Seen, \Answered, \Flagged, \Deleted, and \Draft
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
      */
     public function clearFlag(string $flag, $numbers): bool;
 
@@ -97,8 +97,8 @@ public function getThread(): array;
     /**
      * Bulk move messages.
      *
-     * @param array|MessageIterator|string $numbers Message numbers
-     * @param MailboxInterface             $mailbox Destination Mailbox to move the messages to
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
+     * @param MailboxInterface                      $mailbox Destination Mailbox to move the messages to
      *
      * @throws \Ddeboer\Imap\Exception\MessageMoveException
      */
@@ -107,8 +107,8 @@ public function move($numbers, self $mailbox): void;
     /**
      * Bulk copy messages.
      *
-     * @param array|MessageIterator|string $numbers Message numbers
-     * @param MailboxInterface             $mailbox Destination Mailbox to copy the messages to
+     * @param array|MessageIteratorInterface|string $numbers Message numbers
+     * @param MailboxInterface                      $mailbox Destination Mailbox to copy the messages to
      *
      * @throws \Ddeboer\Imap\Exception\MessageCopyException
      */