Move badge to the correct button