Update DraftBehavior.php