diff --git a/python/mlx/nn/layers/positional_encoding.py b/python/mlx/nn/layers/positional_encoding.py
index 7ed4a19d7..1c586693f 100644
--- a/python/mlx/nn/layers/positional_encoding.py
+++ b/python/mlx/nn/layers/positional_encoding.py
@@ -227,4 +227,4 @@ class ALiBi(Module):
         )
         if mask is not None:
             alibi_mask = alibi_mask + mask
-        return attention_scores + alibi_mask
\ No newline at end of file
+        return attention_scores + alibi_mask