diff --git a/R/Estimator.R b/R/Estimator.R
index 6ba9283..4ef2047 100644
--- a/R/Estimator.R
+++ b/R/Estimator.R
@@ -267,7 +267,7 @@ predictDeepEstimator <- function(plpModel,
   } else {
     prediction$value <- plpModel$model$predict_proba(data)
   }
-
+  prediction$value <- as.numeric(prediction$value)
   attr(prediction, "metaData")$modelType <- attr(plpModel, "modelType")
 
   return(prediction)