WP integration fix