404 для attachment страниц

В случае, если из-за ошибок в сео (например из-за неверных настроек плагина aio) в выдачу попали страницы вложений — нужно настроить для них код ответа 404

/**Редирект со страницы вложения attachment на страницу 404**/
add_action('template_redirect', 'template_redirect_attachment');
function template_redirect_attachment() {
global $post;
if (is_attachment()) {
wp_redirect(get_permalink($post->post_parent));
 status_header( 404 );
        nocache_headers();
        include( get_query_template( '404' ) );
        die();
}
}