diff --git a/programming_examples/basic/matrix_multiplication/cascade/aie2.py b/programming_examples/basic/matrix_multiplication/cascade/aie2.py
index 3358b68d03..51e7f5dddb 100644
--- a/programming_examples/basic/matrix_multiplication/cascade/aie2.py
+++ b/programming_examples/basic/matrix_multiplication/cascade/aie2.py
@@ -230,6 +230,7 @@ def device_body():
                         (t, 1),
                     ],
                 )
+            of_offsets = [n * k * i for i in range(n_aie_cols)]
             object_fifo_link(
                 B_l3l2_fifos[col],
                 [B_l2l1_fifos[row][col] for row in range(n_aie_rows)],