From 3d5661bb1b145ff31f7abda10598f5bde0d000a8 Mon Sep 17 00:00:00 2001
From: xuewei cao <36172337+xueweic@users.noreply.github.com>
Date: Mon, 21 Apr 2025 04:21:08 -0400
Subject: [PATCH] Update colocboost_update.R

---
 R/colocboost_update.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/R/colocboost_update.R b/R/colocboost_update.R
index 24be049..546c7df 100644
--- a/R/colocboost_update.R
+++ b/R/colocboost_update.R
@@ -109,7 +109,7 @@ colocboost_update <- function(cb_model, cb_model_para, cb_data) {
       xtx <- cb_data$data[[X_dict]]$XtX
       cb_model[[i]]$res <- rep(0, cb_model_para$P)
       if (length(cb_data$data[[i]]$variable_miss) != 0) {
-        beta <- beta[-cb_data$data[[i]]$variable_miss]  / beta_scaling
+        beta <- cb_model[[i]]$beta[-cb_data$data[[i]]$variable_miss]  / beta_scaling
         xty <- cb_data$data[[i]]$XtY[-cb_data$data[[i]]$variable_miss]
         cb_model[[i]]$res[-cb_data$data[[i]]$variable_miss] <- xty - scaling_factor * xtx %*% beta
       } else {