I have changed the add_head method to return the html, so added this fix here too.