diff --git a/README.md b/README.md
index 35debfac7..f25465af1 100644
--- a/README.md
+++ b/README.md
@@ -166,7 +166,7 @@ Then install and test Flash Attention with the flag `FLASH_ATTENTION_TRITON_AMD_
 export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
 cd flash-attention
 python setup.py install
-pytest tests/test_flash_attn.py
+pytest tests/test_flash_attn_triton_amd.py
 ```
 
 
diff --git a/flash_attn/flash_attn_triton_amd/README.md b/flash_attn/flash_attn_triton_amd/README.md
index 798d78a12..fc411425f 100644
--- a/flash_attn/flash_attn_triton_amd/README.md
+++ b/flash_attn/flash_attn_triton_amd/README.md
@@ -40,7 +40,7 @@ Then install and test Flash Attention with the flag `FLASH_ATTENTION_TRITON_AMD_
 export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
 cd flash-attention
 python setup.py install
-pytest tests/test_flash_attn.py
+pytest tests/test_flash_attn_triton_amd.py
 ```
 
 #### Credits