diff --git a/Overlap/plot_traj.py b/Overlap/plot_traj.py
index 99e98b1681640d84a008feba306eb0be46282e2a..945134c65d19880410fd5ff5a2b3fcfc26facb4d 100755
--- a/Overlap/plot_traj.py
+++ b/Overlap/plot_traj.py
@@ -8,6 +8,7 @@ import itertools
 import numpy as np
 from matplotlib import patches
 import cartopy.crs as ccrs
+import report_graph
 
 color_iter = itertools.cycle(('#1f77b4', '#aec7e8', '#ff7f0e',
                               '#ffbb78', '#2ca02c', '#98df8a',
@@ -37,7 +38,8 @@ def is_node_type(G, n, label):
     elif label == "important":
         return G.in_degree[n] == 0 or G.out_degree[n] == 0 or G.degree[n] >= 3
     elif isinstance(label, int):
-        return node_to_date_eddy(n, e_overestim, d_init)["date"] == label
+        return report_graph.node_to_date_eddy(n, G.graph["e_overestim"])\
+            ["date_index"] == label
 
 def plot_nbunch(G, nbunch, color = '#1f78b4', label = None, ax = None):
     if ax is None: ax = plt.gca()
@@ -93,7 +95,6 @@ def animate_nbunch(G, nbunch):
         date += 1
     
 if __name__ == "__main__":
-    import report_graph
     import argparse
     import sys
     from os import path