diff --git a/swash/processing/layers.py b/swash/processing/layers.py
index dfe41d5..d1656b4 100644
--- a/swash/processing/layers.py
+++ b/swash/processing/layers.py
@@ -50,7 +50,6 @@ fig, ax = plt.subplots()
 # )
 
 n = 0
-velk = velk.reshape((6001, 10, 2, 1251))
 vk = np.sqrt((velk[n] ** 2).sum(axis=1))
 # print(vk.shape)
 # plt.imshow(vk)
@@ -83,6 +82,7 @@ def animate(k):
         )
     for i, l in enumerate(lines):
         l.set_ydata(zk[k, i])
+    return *quiv, *lines
 
 
 ani = animation.FuncAnimation(