Updated compatibility for WP 3.9