Losses on aperture downstream of collimators should only be shifted to the collimator if it is active.

Author: freddieknets

xcoll/lossmap.py

@@ -254,7 +254,8 @@ def _correct_absorbed(self, part, line, verbose=True):
                     prev_elem = len(line.element_names) - 1
                 else:
                     prev_elem = elem - 1
-                if line.element_names[prev_elem] in coll_elements:
+                if line.element_names[prev_elem] in coll_elements \
+                and line[prev_elem].active:
                     if verbose:
                         print(f"Found at {line.element_names[elem]}, "
                             + f"moved to {line.element_names[elem-1]}")