diff --git a/src/compare_essential_dynamics.py b/src/compare_essential_dynamics.py
index 3fc39f0..f60c332 100755
--- a/src/compare_essential_dynamics.py
+++ b/src/compare_essential_dynamics.py
@@ -135,7 +135,7 @@ def get_energy_basins(pcs, traj, outbasename, marker="o",
      Returns:
      The basin centroids
     """
-    subplot = sns.kdeplot(x=pcs[:, 0], y=pcs[:, 1], shade=True, alpha=0.2,
+    subplot = sns.kdeplot(x=pcs[:, 0], y=pcs[:, 1], fill=True, alpha=0.2,
                           levels=kde_levels, cmap="jet_r")
     basin_paths = subplot.collections[-1].get_paths()
     basins = [basin.vertices for basin in basin_paths]