Corrected problem with badge button