Update for WordPress 3.9 compatibility