From 4cc243b154ea721d6b1df7768104d8d15979f2ff Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Thu, 21 Apr 2022 14:33:01 -0500
Subject: [PATCH] Make sure `diff_burnable_mats` updates materials
 appropriately

---
 openmc/deplete/operator.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/openmc/deplete/operator.py b/openmc/deplete/operator.py
index b5e377bedd..337e8f8c03 100644
--- a/openmc/deplete/operator.py
+++ b/openmc/deplete/operator.py
@@ -84,11 +84,11 @@ class Operator(TransportOperator):
 
     .. versionchanged:: 0.13.0
         The geometry and settings parameters have been replaced with a
-        model parameter that takes an openmc.Model object
+        model parameter that takes a :class:`~openmc.model.Model` object
 
     Parameters
     ----------
-    model : openmc.Model
+    model : openmc.model.Model
         OpenMC model object
     chain_file : str, optional
         Path to the depletion chain XML file.  Defaults to the file
@@ -163,7 +163,7 @@ class Operator(TransportOperator):
 
     Attributes
     ----------
-    model : openmc.Model
+    model : openmc.model.Model
         OpenMC model object
     geometry : openmc.Geometry
         OpenMC geometry object
@@ -195,8 +195,6 @@ class Operator(TransportOperator):
     prev_res : ResultsList or None
         Results from a previous depletion calculation. ``None`` if no
         results are to be used.
-    diff_burnable_mats : bool
-        Whether to differentiate burnable materials with multiple instances
     cleanup_when_done : bool
         Whether to finalize and clear the shared library memory when the
         depletion operation is complete. Defaults to clearing the library.
@@ -248,7 +246,6 @@ class Operator(TransportOperator):
             )
         self.materials = model.materials
 
-        self.diff_burnable_mats = diff_burnable_mats
         self.cleanup_when_done = True
 
         # Reduce the chain before we create more materials
@@ -262,8 +259,11 @@ class Operator(TransportOperator):
             self.chain = self.chain.reduce(all_isotopes, reduce_chain_level)
 
         # Differentiate burnable materials with multiple instances
-        if self.diff_burnable_mats:
+        if diff_burnable_mats:
             self._differentiate_burnable_mats()
+            self.materials = openmc.Materials(
+                model.geometry.get_all_materials().values()
+            )
 
         # Clear out OpenMC, create task lists, distribute
         openmc.reset_auto_ids()