functions.php fix-2