diff --git a/fluidsim/solvers/ns3d/output/spectra.py b/fluidsim/solvers/ns3d/output/spectra.py
index 9b80c0e27161e8837da34aa74225c264fb0cb54d_Zmx1aWRzaW0vc29sdmVycy9uczNkL291dHB1dC9zcGVjdHJhLnB5..ae0dedfbdaa3efadc31b8bbf17bd5e17d5a93e53_Zmx1aWRzaW0vc29sdmVycy9uczNkL291dHB1dC9zcGVjdHJhLnB5 100644
--- a/fluidsim/solvers/ns3d/output/spectra.py
+++ b/fluidsim/solvers/ns3d/output/spectra.py
@@ -225,7 +225,7 @@
         if ylim is not None:
             ax.set_ylim(ylim)
 
-    def plot_kzkh(self, tmin=0, tmax=None, key="K", ax=None):
+    def plot_kzkh(self, tmin=0, tmax=None, key="K", ax=None, vmin=None, vmax=None):
         data = self.load_kzkh_mean(tmin, tmax, key)
         spectrum = np.log10(data[key])
         kz = data["kz"]
@@ -233,6 +233,8 @@
 
         if ax is None:
             fig, ax = self.output.figure_axe()
+        else:
+            fig = ax.figure
 
         ax.set_xlabel(r"$\kappa_h$")
         ax.set_ylabel("$k_z$")
@@ -236,5 +238,5 @@
 
         ax.set_xlabel(r"$\kappa_h$")
         ax.set_ylabel("$k_z$")
-        ax.set_title("log 3D spectra\n" + self.output.summary_simul)
+        ax.set_title("log10 spectra\n" + self.output.summary_simul)
 
@@ -240,5 +242,6 @@
 
-        ax.pcolormesh(kh, kz, spectrum)
+        qmesh = ax.pcolormesh(kh, kz, spectrum, vmin=vmin, vmax=vmax)
+        fig.colorbar(qmesh)
 
     def plot1d(
         self,
@@ -324,5 +327,5 @@
         ks_no0[ks == 0] = np.nan
         coef_norm = ks_no0 ** (coef_compensate)
 
-        style_line = ""
+        style_line = "r"
         if direction == "z":
@@ -328,6 +331,6 @@
         if direction == "z":
-            style_line = ":"
+            style_line = "g"
 
         ax.plot(
             ks,
             spectrum * coef_norm,
@@ -330,8 +333,8 @@
 
         ax.plot(
             ks,
             spectrum * coef_norm,
-            "k" + style_line,
+            style_line,
             linewidth=2,
             label=f"$E(k_{direction})$",
         )