diff --git a/Trajectories/Tests/tests.json b/Trajectories/Tests/tests.json
index b37cc7cd6c55a5168e473f7ef3efad941066d281..58c31ab23cbf36d342162470b1c335745384c4b6 100644
--- a/Trajectories/Tests/tests.json
+++ b/Trajectories/Tests/tests.json
@@ -103,8 +103,7 @@
             "$my_data_dir/Eurec4A/Graph_anti/segments_cost_functions.gt",
             "segments.gv",
             "-t", "$my_data_dir/Eurec4A/Graph_anti/traj_segm.json"
-	],
-	"input": "164\n"
+	]
     },
     "Plot_traj": {
 	"command": [
diff --git a/Trajectories/draw_segments.py b/Trajectories/draw_segments.py
index ce7f1f4d54b606b7ef3aaca7f2761576bfac5b2d..6971b3a48b7113bef4646128d4c0446bbcec2811 100755
--- a/Trajectories/draw_segments.py
+++ b/Trajectories/draw_segments.py
@@ -29,9 +29,7 @@ g_out = pgv.AGraph(directed = True)
 for v in g_in.vertices():
     g_out.add_node(g_in.vp.name[v])
 
-e_overestim = input("Enter e_overestim: ")
-e_overestim = int(e_overestim)
-report_graph.partition_graph_date(g_out, e_overestim)
+report_graph.partition_graph_date(g_out, g_in.gp.e_overestim)
 
 for e in g_in.edges():
     source  = e.source()