diff --git a/cheetah/accelerator/element.py b/cheetah/accelerator/element.py index 4d8ad7d76..70d35084b 100644 --- a/cheetah/accelerator/element.py +++ b/cheetah/accelerator/element.py @@ -92,8 +92,8 @@ def track(self, incoming: Beam) -> Beam: return ParameterBeam( new_mu, new_cov, - incoming.energy, - total_charge=incoming.total_charge, + incoming.energy.clone(), + total_charge=incoming.total_charge.clone(), s=new_s, species=incoming.species.clone(), ) @@ -103,9 +103,9 @@ def track(self, incoming: Beam) -> Beam: new_s = incoming.s + self.length return ParticleBeam( new_particles, - incoming.energy, - particle_charges=incoming.particle_charges, - survival_probabilities=incoming.survival_probabilities, + incoming.energy.clone(), + particle_charges=incoming.particle_charges.clone(), + survival_probabilities=incoming.survival_probabilities.clone(), s=new_s, species=incoming.species.clone(), )