diff --git a/forge/test/mlir/llama/test_llama_train_ops.py b/forge/test/mlir/llama/test_llama_train_ops.py
index 62fa844bc..d2bddfaf6 100644
--- a/forge/test/mlir/llama/test_llama_train_ops.py
+++ b/forge/test/mlir/llama/test_llama_train_ops.py
@@ -67,7 +67,7 @@ def forward(self, x):
 
 
 @pytest.mark.parametrize(
-        "in_features, out_features", [pytest.param(3200, 3200), pytest.param(3200, 8640), pytest.param(8640, 3200)]
+    "in_features, out_features", [pytest.param(3200, 3200), pytest.param(3200, 8640), pytest.param(8640, 3200)]
 )
 @pytest.mark.push
 def test_matmul_dims(in_features, out_features):