Just re-watched (such a great movie) - after Wanda blew up the Mind Stone, Thanos only rolled back what happened to Vision. Wanda was just feet away and didn't rewind, so he can definitely localize it.
I just re-watched the scene where he teleports after doing the snap, it looks like all of the stones are lit up when he leaves. Plus all the colors from the stones form around the axe as he uses the gauntlet. When he puts the final stone into the gauntlet you can see the colors surge into him, it's the same looking thing around that axe, all the colors. Plus even the stones looked damaged.