Update display-functions.php