Edit success flash to make it universal