diff --git a/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py b/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py
index ecf29e6067..2da47efa43 100644
--- a/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py
+++ b/submarine-sdk/pysubmarine/tests/ml/pytorch/model/conftest.py
@@ -70,7 +70,7 @@ def get_model_param(tmpdir):
                 "embedding_dim": 16,
                 "attention_dim": 64,
                 "hidden_units": [400, 400],
-                "dropout_rate": 0.3, 
+                "dropout_rate": 0.3,
                 "dropout_rates": [0.2, 0.2]
             }
         },