diff --git a/tests/test_flash_attn.py b/tests/models/layers/test_flash_attn.py
similarity index 100%
rename from tests/test_flash_attn.py
rename to tests/models/layers/test_flash_attn.py