diff --git a/spinup/utils/plot.py b/spinup/utils/plot.py
index f06d78c41..3824e2d65 100644
--- a/spinup/utils/plot.py
+++ b/spinup/utils/plot.py
@@ -39,7 +39,7 @@ def plot_data(data, xaxis='Epoch', value="AverageEpRet", condition="Condition1",
 
     Changes the colorscheme and the default legend style, though.
     """
-    plt.legend(loc='best').draggable()
+    plt.legend(loc='best').set_draggable(True)
 
     """
     For the version of the legend used in the Spinning Up benchmarking page,