diff --git a/xtrack/loss_location_refinement/loss_location_refinement.py b/xtrack/loss_location_refinement/loss_location_refinement.py index 6a9b6b978..b284f6ea8 100644 --- a/xtrack/loss_location_refinement/loss_location_refinement.py +++ b/xtrack/loss_location_refinement/loss_location_refinement.py @@ -9,14 +9,14 @@ import xobjects as xo import xpart as xp from ..tracker import Tracker -from ..beam_elements import LimitPolygon, XYShift, SRotation, Drift +from ..beam_elements import LimitPolygon, XYShift, SRotation, Drift, Marker from ..line import Line import logging logger = logging.getLogger(__name__) logger.addHandler(logging.StreamHandler()) -_default_allowed_backtrack_types = [Drift, SRotation, XYShift] +_default_allowed_backtrack_types = [Drift, SRotation, XYShift, Marker] class LossLocationRefinement: