diff --git a/modules/pytorch_wrapper/ptn_multihead_attention_custom.py b/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
index 7b4b86c..e410ca0 100644
--- a/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
+++ b/modules/pytorch_wrapper/ptn_multihead_attention_custom.py
@@ -6,7 +6,7 @@ from einops import einops
 
 class PtnMultiheadAttentionCustomDef(nn.Module):
     """
-    Implements MultiheadAttention wrapper.
+    Implements MultiheadAttention.
 
     Attributes:
         embed_dim (int): Hidden dimension (d_model) for each token of the Transformer.