From d49f34a4346237980b747bfcefb0f3f19511cf19 Mon Sep 17 00:00:00 2001
From: Noe Brucy <noe.brucy@cea.fr>
Date: Tue, 18 Jan 2022 13:15:43 +0100
Subject: [PATCH] [plotter] correct a small bug in overlay sink

---
 plotter.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/plotter.py b/plotter.py
index ef28f74..7f27afc 100644
--- a/plotter.py
+++ b/plotter.py
@@ -945,12 +945,11 @@ class Plotter(Aggregator, BaseProcessor):
         if sinks:
             try:
                 self.current_processor.sinks()
-                sinks = pd.DataFrame(
+                data = pd.DataFrame(
                     self.current_processor.get_value("/datasets/sinks")
                 )
-                part_pos = sinks[["x", "y", "z"]].values
+                part_pos = data[["x", "y", "z"]].values
                 unit_length /= self.current_processor.lbox
-                data = sinks
             except KeyError:
                 self.current_processor._log("No sinks particles", "WARNING")
                 return
@@ -1000,8 +999,7 @@ class Plotter(Aggregator, BaseProcessor):
         # Size and color
         if s is None and sinks:
             s = data.msink[mask] / 5e3
-
-        if s in data.keys():
+        elif s in data.keys():
             s = data[s][mask]
         elif callable(s):
             s = s(data)[mask]