Fix tests for WordPress 4.0