Fix displaying badge when adding for the first time