From 8a994b8c199f5108b208fbb5153ecf62d81ec23c Mon Sep 17 00:00:00 2001
From: giadarol <giovanni.iadarola@cern.ch>
Date: Fri, 21 Oct 2022 13:25:22 +0200
Subject: [PATCH] Fix for GPU

---
 tests/test_beambeam3d.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_beambeam3d.py b/tests/test_beambeam3d.py
index f03f74c2..dfcd57b1 100644
--- a/tests/test_beambeam3d.py
+++ b/tests/test_beambeam3d.py
@@ -619,6 +619,7 @@ def test_beambeam3d_old_interface():
              # Scaling down bb:
             bb.scale_strength = 0
             part_before_tracking = part.copy()
+            part.move(_context=context)
             bb.track(part)
 
             for cc in 'x px y py zeta delta'.split():