diff --git a/decoder/ih264d_utils.c b/decoder/ih264d_utils.c
index e65a939..170bc97 100644
--- a/decoder/ih264d_utils.c
+++ b/decoder/ih264d_utils.c
@@ -2195,9 +2195,8 @@ WORD16 ih264d_get_memory_dec_params(dec_struct_t * ps_dec)
         u4_scratch_mem_used = ALIGN64(u4_scratch_mem_used);
         ps_dec->pu4_mbaff_wt_mat = (UWORD32 *)(pu1_scratch_mem_base + u4_scratch_mem_used);
 
-        u4_scratch_mem_used += (sizeof(UWORD32) * 3
-                        * (MAX_FRAMES * MAX_FRAMES))
-                        << 3;
+        u4_scratch_mem_used += (sizeof(UWORD32) * 2 * 3 *
+                        ((MAX_FRAMES << 1) * (MAX_FRAMES << 1)) * 2);
         u4_scratch_mem_used = ALIGN64(u4_scratch_mem_used);
 
         ps_dec->pu4_wts_ofsts_mat = (UWORD32 *)(pu1_scratch_mem_base + u4_scratch_mem_used);