Update to work with WP 3.8.1