We should probably create our own loot table for this. I believe we are using the End City loot table which is too much, but I think most other existing loot tables are too weak. Maybe the Nether Fortress one or Stronghold would be sufficient, but I don't think that loot is good enough.