diff --git a/script/Alignment/Gun_solenoid_alignment_dv_sim.py b/script/Alignment/Gun_solenoid_alignment_dv_sim.py
index b48d520..1819203 100644
--- a/script/Alignment/Gun_solenoid_alignment_dv_sim.py
+++ b/script/Alignment/Gun_solenoid_alignment_dv_sim.py
@@ -25,9 +25,8 @@ for m in mag:
 
 r = lscan(I_set, [I_read, bpm_x, bpm_y], I1, I2, dI)
 
-x = r.getReadable(1)
-y = r.getReadable(2)
-plot(y, xdata=x, title = "Centroid excursion")
+plot(r.getReadable(2), xdata=r.getReadable(1), title = "Centroid excursion")
+set_return(r.print())
 
 msg = str(r)
 msg = msg + "\nFile: " + get_context().path