diff --git a/R/R/inputs.R b/R/R/inputs.R
index b7d117633..0da5429db 100644
--- a/R/R/inputs.R
+++ b/R/R/inputs.R
@@ -721,12 +721,11 @@ robyn_engineering <- function(x, quiet = FALSE, ...) {
   InputCollect[["dt_mod"]] <- dt_transform
   InputCollect[["dt_modRollWind"]] <- dt_transform[rollingWindowStartWhich:rollingWindowEndWhich, ]
   InputCollect[["dt_inputRollWind"]] <- dt_inputRollWind
-  InputCollect[["modNLSCollect"]] <- modNLSCollect
-  InputCollect[["plotNLSCollect"]] <- plotNLSCollect
-  InputCollect[["yhatNLSCollect"]] <- yhatNLSCollect
-
-  # InputCollect[["exposure_selector"]] <- exposure_selector
-  # InputCollect[["mediaCostFactor"]] <- mediaCostFactor
+  InputCollect[["modNLS"]] <- list(
+    results = modNLSCollect,
+    yhat = yhatNLSCollect,
+    plots = plotNLSCollect
+  )
   return(InputCollect)
 }
 
diff --git a/demo/demo.R b/demo/demo.R
index 055d03458..91780e16a 100644
--- a/demo/demo.R
+++ b/demo/demo.R
@@ -271,8 +271,8 @@ print(InputCollect)
 
 #### Check spend exposure fit if available
 if (length(InputCollect$exposure_vars) > 0) {
-  InputCollect$plotNLSCollect$facebook_I
-  InputCollect$plotNLSCollect$search_clicks_P
+  InputCollect$modNLS$plots$facebook_I
+  InputCollect$modNLS$plots$search_clicks_P
 }
 
 ##### Manually save and import InputCollect as JSON file