diff --git a/plotting.py b/plotting.py
index fbc1c775e3f5fdb2578cef307179b39b925c2639..0d9de10e01f8a56eea3f5d00d9e4395246429198 100644
--- a/plotting.py
+++ b/plotting.py
@@ -47,7 +47,7 @@ def kymograph(
             cmap = matplotlib.cm.Spectral
         else:
             cmap = matplotlib.cm.viridis
-    wdf = df.pivot(y, x, hue)
+    wdf = df.pivot(index=y, columns=x, values=hue)
     dt = np.min(np.diff(np.sort(df.time.unique())))
     # from Arin
     data = wdf.to_numpy()