diff --git a/examples/Project.toml b/examples/Project.toml
index 61c9d6c685f..c89bea32734 100644
--- a/examples/Project.toml
+++ b/examples/Project.toml
@@ -67,7 +67,7 @@ YAML = "ddb6d928-2868-570f-bddf-ab3f9cf99eb6"
 [compat]
 BlockArrays = "0.16"
 CairoMakie = "0.10, 0.11"
-ClimaAnalysis = "0.5.1"
+ClimaAnalysis = "0.5.7"
 ClimaCoreMakie = "0.3, 0.4"
 ClimaCorePlots = "0.2"
 ClimaCoreSpectra = "0.1"
diff --git a/post_processing/ci_plots.jl b/post_processing/ci_plots.jl
index 148b9775bef..8943904ef0a 100644
--- a/post_processing/ci_plots.jl
+++ b/post_processing/ci_plots.jl
@@ -1107,10 +1107,11 @@ them on the same axis.
 """
 function plot_edmf_vert_profile!(grid_loc, var_group)
     z = var_group[1].dims["z"]
-    units = var_group[1].attributes["units"]
+    units = ClimaAnalysis.units(var_group[1])
+    z_units = ClimaAnalysis.dim_units(var_group[1], "z")
     ax = CairoMakie.Axis(
         grid_loc[1, 1],
-        ylabel = "z [$(var_group[1].dim_attributes["z"]["units"])]",
+        ylabel = "z [$z_units]",
         xlabel = "$(short_name(var_group[1])) [$units]",
         title = parse_var_attributes(var_group[1]),
     )