(function moveTdModalClass() {
  document.querySelectorAll('.wp-block-image div.td-modal-image').forEach(function(div) {
    const link = div.querySelector('a');

    if (link && !link.classList.contains('td-modal-image')) {
      link.classList.add('td-modal-image');
      div.classList.remove('td-modal-image');
    }
  });

  setTimeout(moveTdModalClass, 500);
})();