diff --git a/modules/Visualization/R/plot_extreme_probs.R b/modules/Visualization/R/plot_extreme_probs.R
index 1883f96eb36f228b94326f464aef265789f87e23..49cd817510825c722f10dd191d82e3044f302664 100644
--- a/modules/Visualization/R/plot_extreme_probs.R
+++ b/modules/Visualization/R/plot_extreme_probs.R
@@ -91,9 +91,10 @@ plot_extreme_probs <- function(recipe, fcst,
       # Loop over bin dimension
       for (bin in 1:dim(probs_array)["bin"]) {
         variable <- fcst$attrs$Variable$varName[[var]]
+        custom_dictionary <- recipe$Analysis$Workflow$Visualization$custom_dictionary
         var_long_name <- .get_variable_longname(data_cube = fcst,
                                                 index = var,
-                                                format = NULL) ## TODO: change
+                                                format = custom_dictionary)
         # Subset ensemble mean by variable
         var_fcst <- ClimProjDiags::Subset(probs_array,
                                           along = c("bin", "var"),
diff --git a/modules/Visualization/R/plot_forecast_map.R b/modules/Visualization/R/plot_forecast_map.R
index f683f704bd183277bdcda1fa26408f4f93dd5eac..892f0fda943f83b41003577118e4780ff35f0e69 100644
--- a/modules/Visualization/R/plot_forecast_map.R
+++ b/modules/Visualization/R/plot_forecast_map.R
@@ -78,7 +78,7 @@ plot_forecast_map <- function(recipe, fcst, mask = NULL, dots = NULL,
   # Loop over variable dimension
   for (var in 1:fcst$dims[['var']]) {
     variable <- fcst$attrs$Variable$varName[[var]]
-    custom_dictionary <- recipe$Analysis$Visualization$custom_dictionary
+    custom_dictionary <- recipe$Analysis$Workflow$Visualization$custom_dictionary
     var_long_name <- .get_variable_longname(data_cube = fcst,
                                             index = var,
                                             format = custom_dictionary)
diff --git a/modules/Visualization/R/plot_metrics.R b/modules/Visualization/R/plot_metrics.R
index 1b1cd5b52b1319af6601af0aad214639d62cb878..7bb064b3c6ce8793b53272c9749fbdff7e60cbf9 100644
--- a/modules/Visualization/R/plot_metrics.R
+++ b/modules/Visualization/R/plot_metrics.R
@@ -257,7 +257,7 @@ plot_metrics <- function(recipe, data_cube, metrics,
         
         # Get variable name and long name
         var_name <- data_cube$attrs$Variable$varName[[var]]
-        custom_dictionary <- recipe$Analysis$Visualization$custom_dictionary
+        custom_dictionary <- recipe$Analysis$Workflow$Visualization$custom_dictionary
         var_long_name <- .get_variable_longname(data_cube = data_cube,
                                                 index = var,
                                                 format = custom_dictionary)
diff --git a/modules/Visualization/R/plot_most_likely_terciles_map.R b/modules/Visualization/R/plot_most_likely_terciles_map.R
index e7bdbc25e8660f09d88c7d9e0035be555d4fd3df..cf9ad7adf2dc7cac0afffeae9c30a4173da761b3 100644
--- a/modules/Visualization/R/plot_most_likely_terciles_map.R
+++ b/modules/Visualization/R/plot_most_likely_terciles_map.R
@@ -95,7 +95,7 @@ plot_most_likely_terciles <- function(recipe,
   for (var in 1:fcst$dims[['var']]) {
     ## NOTE: Variables starting with var_* represent the variable counter.
     variable <- fcst$attrs$Variable$varName[[var]]
-    custom_dictionary <- recipe$Analysis$Visualization$custom_dictionary
+    custom_dictionary <- recipe$Analysis$Workflow$Visualization$custom_dictionary
     var_long_name <- .get_variable_longname(data_cube = fcst,
                                             index = var,
                                             format = custom_dictionary)