diff --git a/src/cristallina/plot.py b/src/cristallina/plot.py
index 4304f1a..c7e11e9 100644
--- a/src/cristallina/plot.py
+++ b/src/cristallina/plot.py
@@ -86,7 +86,7 @@ def plot_channel(data, channel_name, ax=None):
         3: plot_image_channel,
     }
 
-    plot_f[channel_dim](data, channel_name, ax)
+    plot_f[channel_dim](data, channel_name, ax=ax)
 
 
 def axis_styling(ax, channel_name, description):