Update src/predict.jl

Co-authored-by: Alex Arslan <[email protected]>
JuliaStats · Mar 25, 2024 · 869e0de · 869e0de
1 parent 729dbcb
commit 869e0de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/predict.jl b/src/predict.jl
@@ -126,7 +126,7 @@ function _predict(m::MixedModel{T}, newdata, β; new_re_levels) where {T}
         ytemp, lmm
     end
 
-    pivotmatch = mnew.feterm.piv[pivot(m)]
+    pivotmatch = pivot(mnew)[pivot(m)]
     grps = fnames(m)
     mul!(y, view(mnew.X, :, pivotmatch), β)
     # mnew.reterms for the correct Z matrices