Update DoctrineTrait.php