diff --git a/av1/encoder/encodemb.c b/av1/encoder/encodemb.c
index 6b0991f..c441421 100644
--- a/av1/encoder/encodemb.c
+++ b/av1/encoder/encodemb.c
@@ -906,13 +906,9 @@
 #endif  // !CONFIG_PVQ
 #if CONFIG_HIGHBITDEPTH
     if (xd->cur_buf->flags & YV12_FLAG_HIGHBITDEPTH) {
-      if (xd->lossless[xd->mi[0]->mbmi.segment_id]) {
-        av1_highbd_iwht4x4_add(dqcoeff, dst, pd->dst.stride, p->eobs[block],
-                               xd->bd);
-      } else {
-        av1_highbd_idct4x4_add(dqcoeff, dst, pd->dst.stride, p->eobs[block],
-                               xd->bd);
-      }
+      av1_highbd_inv_txfm_add_4x4(dqcoeff, dst, pd->dst.stride, p->eobs[block],
+                                  xd->bd, DCT_DCT,
+                                  xd->lossless[xd->mi[0]->mbmi.segment_id]);
       return;
     }
 #endif  //  CONFIG_HIGHBITDEPTH