Small change to functions.php