Fix for functions.php