Positive reinforcement works better than negative. If noticing is followed by a punishment you are disincentivizing yourself to notice. This is bad because noticing is its own super power. Instead maybe try congratulating yourself for noticing, and then replacing the negative habit with some other reward. Eating too many gummy bears in the short term is probably worth it to repair bad habits in the long term for instance.