Fix DJ add/remove logic