diff --git a/src/search.c b/src/search.c
index b181d5fe..6ce6b9a4 100644
--- a/src/search.c
+++ b/src/search.c
@@ -592,6 +592,9 @@ static double cu_rd_cost_tr_split_accurate(const encoder_state_t* const state,
           chroma_width);
         chroma_ssd = ssd_u_joint + ssd_v_joint;
       }
+      if (chroma_can_use_tr_skip) {
+        CABAC_FBITS_UPDATE(cabac, &cabac->ctx.transform_skip_model_chroma, tr_cu->tr_skip & 2, tr_tree_bits, "transform_skip_flag");
+      }
       coeff_bits += uvg_get_coeff_cost(state, &lcu->coeff.joint_uv[index], NULL, chroma_width, COLOR_U, scan_order, 0);
     }
   }