Fix flash style removal todo

I saw this pop up once at the Surrey tech day yesterday, now it actually removes the style.

Because of this change, you can trigger the highlight multiple times now.