diff --git a/lib/Tmdb/Model/Tv/Season.php b/lib/Tmdb/Model/Tv/Season.php index 2941f38e..cc28db4b 100644 --- a/lib/Tmdb/Model/Tv/Season.php +++ b/lib/Tmdb/Model/Tv/Season.php @@ -76,7 +76,7 @@ class Season extends AbstractModel */ protected $changes; /** - * @var DateTime + * @var \DateTime|null */ private $airDate; /** @@ -118,7 +118,7 @@ public function __construct() } /** - * @return DateTime + * @return ?DateTime */ public function getAirDate() { @@ -126,12 +126,18 @@ public function getAirDate() } /** - * @param string $airDate + * @param DateTime|string|null $airDate * @return self */ - public function setAirDate($airDate) + public function setAirDate($airDate = null) { - $this->airDate = new DateTime($airDate); + if (empty($airDate)) { + $airDate = null; + } elseif (!$airDate instanceof DateTime) { + $airDate = new DateTime($airDate); + } + + $this->airDate = $airDate; return $this; }