Remove Badge model reference