move callback `classRemove` outside the loop

Hi,
Since not able to create the branch for the latest version Drupal 11.x.

So Attaching the rerolled patch file for the core version 11.x.

Created MR from #12 patch against Drupal core 11.x branch.