function.php fix