Tallys of Lattice Elements in 0.14.0

I updated to the new version recently and am having some trouble getting tallies to work using the updated “model.differentiate_depletable_mats”. I have a pincell lattice that I have differentiated, and I want to be able to take tallies on each of the differentiated pins separately.

Current Tally Setup

fiss_filter_cell = openmc.CellFilter([fuel_rod.id])
fiss_tally = openmc.Tally()
fiss_tally.filters = [fiss_filter_cell]
fiss_tally.scores = [‘fission’, ‘kappa-fission’]
tallies = openmc.Tallies([fiss_tally])
tallies.export_to_xml()

Perhaps related to this thread Problem with diff_burnable_mats=True - #3 by Artur

Solution is to use the DistribcellFilter instead of CellFilter like so:
distrib_filter = openmc.DistribcellFilter(fuel_rod)
Where “fuel_rod” is the name of my fuel pin cell.