diff --git a/llms/mlx_lm/models/olmo2.py b/llms/mlx_lm/models/olmo2.py
index f8efda14..a28fdcc1 100644
--- a/llms/mlx_lm/models/olmo2.py
+++ b/llms/mlx_lm/models/olmo2.py
@@ -231,7 +231,9 @@ class TransformerBlock(nn.Module):
         self.post_attention_layernorm = nn.RMSNorm(
             args.hidden_size, eps=args.rms_norm_eps
         )
-        self.post_feedforward_layernorm = nn.RMSNorm(args.hidden_size, eps=args.rms_norm_eps)
+        self.post_feedforward_layernorm = nn.RMSNorm(
+            args.hidden_size, eps=args.rms_norm_eps
+        )
         self.args = args
 
     def __call__(
diff --git a/llms/tests/test_models.py b/llms/tests/test_models.py
index 2d5ed502..edb594d7 100644
--- a/llms/tests/test_models.py
+++ b/llms/tests/test_models.py
@@ -813,6 +813,5 @@ class TestModels(unittest.TestCase):
         )
 
 
-
 if __name__ == "__main__":
     unittest.main()