fix to functions.php