If he is stimming, you can stop one stimulant, but another will take it's place. My son's stimming is much better and under more control now that he is older, however he was never ever punished for his stims. HOWEVER, he did try very hard to hold in vocal stims in public.
Stimming is not a behavior problem.