diff --git a/flash_attn/ops/triton/layernorm.py b/flash_attn/ops/triton/layer_norm.py
similarity index 100%
rename from flash_attn/ops/triton/layernorm.py
rename to flash_attn/ops/triton/layer_norm.py
diff --git a/tests/ops/triton/test_layer_norm.py b/tests/ops/triton/test_layer_norm.py
index bb5f03e..3d92b6b 100644
--- a/tests/ops/triton/test_layer_norm.py
+++ b/tests/ops/triton/test_layer_norm.py
@@ -5,7 +5,7 @@ import torch
 import torch.nn.functional as F
 from einops import rearrange, repeat
 
-from flash_attn.ops.triton.layernorm import (
+from flash_attn.ops.triton.layer_norm import (
     layer_norm_fn,
     layer_norm_ref,
     rms_norm_ref,