Как изменить обертку у изображений в посте?

Нужно изменить стандартную обертку при выводе изображений в посте. Изображения там оборачиваются в <p></p>

На нашел вариант удаления обертки:

add_filter('the_content', 'remove_img_ptags_func');
function remove_img_ptags_func( $content ){
  return preg_replace('/<p>\s*((?:<a[^>]+>)?\s*<img[^>]+>\s*(?:<\/a>)?)\s*<\/p>/i', '\1', $content );
}

Пытался адаптировать под свои цели, но не выходит (в регулярках не силен):

add_filter('the_content', 'remove_img_ptags_func');
function remove_img_ptags_func( $content ){
  $pattern = "/<p>\s*((?:<a[^>]+>)?\s*<img[^>]+>\s*(?:<\/a>)?)\s*<\/p>/i";
  $replacement = "/<div class='wrapper-img'>\s*((?:<a[^>]+>)?\s*<img[^>]+>\s*(?:<\/a>)?)\s*<\/p>/i";
  $content = preg_replace($pattern, $replacement, $content);
  return $content;

}

Подскажите, есть ли возможность поменять обертку у изображений.