Update WithRelatedBehavior.php