diff --git a/examples/pincell_depletion/restart_depletion.py b/examples/pincell_depletion/restart_depletion.py
index fd0a4a8dde..991120399d 100644
--- a/examples/pincell_depletion/restart_depletion.py
+++ b/examples/pincell_depletion/restart_depletion.py
@@ -62,10 +62,10 @@ results = openmc.deplete.ResultsList.from_hdf5("depletion_results.h5")
 time, keff = results.get_eigenvalue()
 
 # Obtain U235 concentration as a function of time
-time, n_U235 = results.get_atoms('1', 'U235')
+time, n_U235 = results.get_atoms(uo2, 'U235')
 
 # Obtain Xe135 capture reaction rate as a function of time
-time, Xe_capture = results.get_reaction_rate('1', 'Xe135', '(n,gamma)')
+time, Xe_capture = results.get_reaction_rate(uo2, 'Xe135', '(n,gamma)')
 
 ###############################################################################
 #                            Generate plots
diff --git a/examples/pincell_depletion/run_depletion.py b/examples/pincell_depletion/run_depletion.py
index 45dfd2a875..75bc3b542e 100644
--- a/examples/pincell_depletion/run_depletion.py
+++ b/examples/pincell_depletion/run_depletion.py
@@ -107,10 +107,10 @@ results = openmc.deplete.ResultsList.from_hdf5("depletion_results.h5")
 time, keff = results.get_eigenvalue()
 
 # Obtain U235 concentration as a function of time
-time, n_U235 = results.get_atoms('1', 'U235')
+time, n_U235 = results.get_atoms(uo2, 'U235')
 
 # Obtain Xe135 capture reaction rate as a function of time
-time, Xe_capture = results.get_reaction_rate('1', 'Xe135', '(n,gamma)')
+time, Xe_capture = results.get_reaction_rate(uo2, 'Xe135', '(n,gamma)')
 
 ###############################################################################
 #                            Generate plots