diff --git a/gemma/flash_attention_test.cc b/gemma/flash_attention_test.cc
index 8231b0c5..33702b88 100644
--- a/gemma/flash_attention_test.cc
+++ b/gemma/flash_attention_test.cc
@@ -190,7 +190,7 @@ HWY_AFTER_NAMESPACE();
 
 namespace gcpp {
 HWY_BEFORE_TEST(FlashAttentionTest);
-// HWY_EXPORT_AND_TEST_P(FlashAttentionTest, TestAttention);
+HWY_EXPORT_AND_TEST_P(FlashAttentionTest, TestAttention);
 HWY_AFTER_TEST();
 
 }  // namespace gcpp