Minor modification for WordPress 4.0