diff --git a/R/R/diagnostics.R b/R/R/diagnostics.R
index 3633a18..e0b9b31 100644
--- a/R/R/diagnostics.R
+++ b/R/R/diagnostics.R
@@ -101,7 +101,7 @@ cross_validation <- function(
   }
   
   predict_columns <- c('ds', 'yhat')
-  if (model$uncertainty_samples){
+  if (model$uncertainty.samples){
     predict_columns <- append(predict_columns, c('yhat_lower', 'yhat_upper'))
   }