correct stupid error

moldyn · Jul 2, 2024 · 830d847 · 830d847
1 parent a9ca918
commit 830d847
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dcTMD/utils/plotting.py b/src/dcTMD/utils/plotting.py
@@ -104,8 +104,8 @@ def plot_Gamma(x, friction, ax, label=None):
 def plot_dG(x, dG, ax, label=None):
     """Plot free energy vs position."""
     if label is None:
-        line, _ = ax.plot(x, dG)
-    line, _ = ax.plot(x, dG, label=label)
+        line, = ax.plot(x, dG)
+    line, = ax.plot(x, dG, label=label)
     ax.set(xlabel=r'position $x$ [nm]',
            ylabel=r'$\Delta G$ [kJ/mol]',
            xlim=[min(x), max(x)],