diff --git a/R/DESCRIPTION b/R/DESCRIPTION
index a641cf785..0d680db47 100644
--- a/R/DESCRIPTION
+++ b/R/DESCRIPTION
@@ -1,7 +1,7 @@
 Package: Robyn
 Type: Package
 Title: Semi-Automated Marketing Mix Modeling (MMM) from Meta Marketing Science 
-Version: 3.12.0.9003
+Version: 3.12.0.9004
 Authors@R: c(
     person("Gufeng", "Zhou", , "gufeng@meta.com", c("cre", "aut")),
     person("Igor", "Skokan", , "igorskokan@meta.com", c("aut")),
diff --git a/R/R/plots.R b/R/R/plots.R
index 37e62b02b..74d827b96 100644
--- a/R/R/plots.R
+++ b/R/R/plots.R
@@ -174,7 +174,7 @@ robyn_plots <- function(
     if (length(temp_all) > 0) {
       xDecompAgg <- temp_all$xDecompAgg
       dt_ridges <- xDecompAgg %>%
-        filter(.data$rn %in% InputCollect$paid_media_spends) %>%
+        filter(.data$rn %in% InputCollect$paid_media_vars) %>%
         mutate(iteration = (.data$iterNG - 1) * OutputCollect$cores + .data$iterPar) %>%
         select(variables = .data$rn, .data$roi_total, .data$iteration, .data$trial) %>%
         arrange(.data$iteration, .data$variables)
@@ -317,7 +317,7 @@ robyn_onepagers <- function(
 
     plotMediaShare <- filter(
       xDecompAgg, .data$robynPareto == pf,
-      .data$rn %in% InputCollect$paid_media_spends
+      .data$rn %in% InputCollect$paid_media_vars
     )
     uniqueSol <- unique(plotMediaShare$solID)