From 570b00196a3cd4f9490ff8760988745e7fe2b40c Mon Sep 17 00:00:00 2001
From: Eltos <eltos@outlook.de>
Date: Thu, 30 Jan 2025 17:23:59 +0100
Subject: [PATCH] Fix test

---
 tests/test_line.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_line.py b/tests/test_line.py
index a6263d3..f94d5fe 100644
--- a/tests/test_line.py
+++ b/tests/test_line.py
@@ -81,9 +81,10 @@ def test_repeated_elements_survey():
     # https://github.com/xsuite/xplt/issues/31
     line = xt.Line(elements={"obm": xt.Bend(length=0.5)}, element_names=["obm", "obm"])
     # line.replace_all_repeated_elements()
-    print(line.survey())
+    survey = line.survey()
+    print(survey)
 
-    plot = xplt.FloorPlot(line=line)
+    plot = xplt.FloorPlot(survey)
 
     boxes = plot.artists_boxes
     assert_equal(len(boxes), 2)