diff --git a/src/partition.php b/src/partition.php
index 4c348d3..fdbeb4e 100644
--- a/src/partition.php
+++ b/src/partition.php
@@ -11,7 +11,7 @@
  * @since 0.1
  *
  * @param int             $size
- * @param array|Generator $coll
+ * @param array|Traversable|Generator $coll
  *
  * @return array
  */