diff --git a/src/JsonSchema.php b/src/JsonSchema.php
index 5e2af72..80b4a89 100644
--- a/src/JsonSchema.php
+++ b/src/JsonSchema.php
@@ -648,7 +648,7 @@ public function setEnum($enum)
     /** @codeCoverageIgnoreEnd */
 
     /**
-     * @param array $type
+     * @param array|string $type
      * @return $this
      * @codeCoverageIgnoreStart
      */