Updated tests for WP patch