diff --git a/tests/regression/flash_attention/kernel.gemmini.cpp b/tests/regression/flash_attention/kernel.gemmini.cpp
index 9208e863..b97e22cf 100644
--- a/tests/regression/flash_attention/kernel.gemmini.cpp
+++ b/tests/regression/flash_attention/kernel.gemmini.cpp
@@ -615,6 +615,7 @@ void kernel_body(int task_id, kernel_arg_t *__UNIFORM__ arg) {
         //   vx_fence();
         // }
         // asm volatile("rescale_fence_read_end_%=:" ::);
+        virgo_fence(2);
 #endif
 
 #if 0